heimoshuiyu/whisper-fastapi
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update model loading message

Browse Source
This commit is contained in:
黑墨水鱼
2024-08-08 17:20:52 +08:00
parent f71ef945db
commit 042800721d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
whisper_fastapi.py
View File

@@ -39,7 +39,7 @@ app = FastAPI()
Instrumentator().instrument(app).expose(app, endpoint="/konele/metrics") Instrumentator().instrument(app).expose(app, endpoint="/konele/metrics")
ccc = opencc.OpenCC("t2s.json") ccc = opencc.OpenCC("t2s.json")
print("Loading model...") print(f"Loading model to device {args.device}...")
transcriber = Transcribe( transcriber = Transcribe(
model_path=args.model, model_path=args.model,
device=args.device, device=args.device,
@@ -49,7 +49,7 @@ transcriber = Transcribe(
cache_directory=args.cache_dir, cache_directory=args.cache_dir,
local_files_only=args.local_files_only, local_files_only=args.local_files_only,
) )
print("Model loaded!") print(f"Model loaded to device {transcriber.model.model.device}")
# allow all cors # allow all cors