Update app.py

app.py

@@ -3,29 +3,25 @@ import gradio as gr
 import os, json
 from loguru import logger
 import random
-import whisper
+from transformers import pipeline
 import torch
 
 session_token = os.environ.get('SessionToken')      
 # logger.info(f"session_token_: {session_token}")
 
-whisper_model = whisper.load_model("medium")
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
-def transcribe(audio):
-    audio = whisper.load_audio(audio)
-    audio = whisper.pad_or_trim(audio)
-
-    mel = whisper.log_mel_spectrogram(audio).to(whisper_model.device)
+whisper_model = pipeline(
+    task="automatic-speech-recognition",
+    model="openai/whisper-large-v2",
+    chunk_length_s=30,
+    device=device,
+)
 
-    _, probs = whisper_model.detect_language(mel)
-    translate_options = whisper.DecodingOptions(task="translate", fp16 = False)
-
-    translation = whisper.decode(whisper_model, mel, translate_options)
-  
-    print("Translated: " + translation.text)
-    return translation.text
+def transcribe(audio):
+    text = whisper_model(audio)["text"]
+    return text
     
 def get_response_from_chatbot(text):
     try: