Update app.py

app.py

@@ -4,6 +4,7 @@ import spaces
 import torch
 
 from transformers import pipeline
+from faster_whisper import WhisperModel
 
 # Pre-Initialize
 DEVICE = "auto"
@@ -15,7 +16,9 @@ print(f"[SYSTEM] | Using {DEVICE} type compute device.")
 DEFAULT_TASK = "transcribe"
 BATCH_SIZE = 8
 
-repo = pipeline(task="automatic-speech-recognition", model="deepdml/faster-whisper-large-v3-turbo-ct2", chunk_length_s=30, device=DEVICE)
+# repo = pipeline(task="automatic-speech-recognition", model="deepdml/faster-whisper-large-v3-turbo-ct2", chunk_length_s=30, device=DEVICE)
+
+repo = WhisperModel("faster-whisper-large-v3-turbo-ct2")
 
 css = '''
 .gradio-container{max-width: 560px !important}
@@ -29,8 +32,12 @@ footer {
 def transcribe(input=None, task=DEFAULT_TASK):
     print(input)
     if input is None: raise gr.Error("Invalid input.")
-    output = repo(input, batch_size=BATCH_SIZE, generate_kwargs={"task": task}, return_timestamps=True)["text"]
-    return output
+        
+    segments, info = model.transcribe(input)
+    print(segments)
+    print(info)
+    # output = repo(input, batch_size=BATCH_SIZE, generate_kwargs={"task": task}, return_timestamps=True)["text"]
+    return segments
 
 def cloud():
     print("[CLOUD] | Space maintained.")