vlad.os/simple-asr-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
vladislav
2025-07-15 02:13:35 +03:00
parent aa1d21f9bc
commit e7f7120f27
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
app.py
View File

@@ -2,10 +2,10 @@ import logging
import os import os
import subprocess import subprocess
import time import time
from os import getenv
from typing import Dict 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 import FastAPI, Depends, HTTPException, UploadFile, File
from fastapi.security import APIKeyHeader from fastapi.security import APIKeyHeader
@@ -18,7 +18,7 @@ logger = logging.getLogger(__name__)
app = FastAPI() 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
api_key_header = APIKeyHeader(name="x-api-key") api_key_header = APIKeyHeader(name="x-api-key")