не надо модель загружать в начале, а то докер треснет
This commit is contained in:
11
app.py
11
app.py
@@ -18,17 +18,6 @@ logger = logging.getLogger(__name__)
|
|||||||
|
|
||||||
app = FastAPI()
|
app = FastAPI()
|
||||||
|
|
||||||
# Load model on startup
|
|
||||||
model = None
|
|
||||||
|
|
||||||
|
|
||||||
@app.on_event("startup")
|
|
||||||
def load_model():
|
|
||||||
global model
|
|
||||||
logger.info("Loading whisper model...")
|
|
||||||
model = whisper.load_model("turbo", device="cuda", in_memory=True)
|
|
||||||
logger.info("Whisper model loaded.")
|
|
||||||
|
|
||||||
|
|
||||||
# API key header
|
# API key header
|
||||||
api_key_header = APIKeyHeader(name="x-api-key")
|
api_key_header = APIKeyHeader(name="x-api-key")
|
||||||
|
|||||||
Reference in New Issue
Block a user