From 042800721da06f1d8ee63b2f30aa3f31f7c2505c Mon Sep 17 00:00:00 2001
From: heimoshuiyu <heimoshuiyu@gmail.com>
Date: Thu, 8 Aug 2024 17:20:52 +0800
Subject: [PATCH] Update model loading message

---
 whisper_fastapi.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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