Делаем HIP совместимость

This commit is contained in:
2025-07-15 02:13:35 +03:00
parent aa1d21f9bc
commit e7f7120f27

6
app.py
View File

@@ -2,10 +2,10 @@ import logging
import os
import subprocess
import time
from os import getenv
from typing import Dict
from typing import Optional, Union, List, Tuple
import gigaam
import gigaam
from fastapi import FastAPI, Depends, HTTPException, UploadFile, File
from fastapi.security import APIKeyHeader
@@ -18,7 +18,7 @@ logger = logging.getLogger(__name__)
app = FastAPI()
model = gigaam.load_model("v2_ctc", device="cuda", download_root="./model")
model = gigaam.load_model("v2_ctc", device=getenv("ASR_DEVICE"), download_root=getenv("ASR_MODELS_ROOT"))
# API key header
api_key_header = APIKeyHeader(name="x-api-key")