minor fix

libs/transformer/get_transcript_2.py

@@ -1,7 +1,10 @@
+import torch
 from transformers import pipeline
 
-def get_transcribe_transformers(url:str):
-    pipe = pipeline("automatic-speech-recognition", model="distil-whisper/distil-large-v3", return_timestamps=True)
+def get_transcribe_transformers(url:str, model: str):
+    device = "cuda:0" if torch.cuda.is_available() else "cpu"
+
+    pipe = pipeline("automatic-speech-recognition", model=model, return_timestamps=True, device=device)
 
     result = pipe(url)
 

routers/get_transcript_transformer.py

@@ -25,7 +25,7 @@ def get_transcript(audio_path: str, model_size: str = "distil-whisper/distil-sma
     convert_to_audio(audio_path.strip(), output_file)
 
     try:
-        text, chunks = get_transcript_gpu(output_file, model_size)
+        text, chunks = get_transcribe_transformers(output_file, model_size)
     except Exception as error:
         raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=f"error>>>: {error}")
     finally:
@@ -34,8 +34,6 @@ def get_transcript(audio_path: str, model_size: str = "distil-whisper/distil-sma
 
     listSentences = []
 
-    print(chunks)
-
     for chunk in chunks:
         listSentences.append({
             "start_time": chunk.get("timestamp")[0],