fix issue

app.py

@@ -1,18 +1,33 @@
 from transformers import pipeline
 import gradio as gr
 import pytube as pt
+import torch
 
-model = pipeline(model="openai/whisper-medium")  # change to "your-username/the-name-you-picked"
+MODEL_NAME = "openai/whisper-medium"
+device = 0 if torch.cuda.is_available() else "cpu"
 
+transcribe = pipeline(
+    task="automatic-speech-recognition",
+    model=MODEL_NAME,
+    chunk_length_s=30,
+    device=device,
+)
+transcribe.model.config.forced_decoder_ids = transcribe.tokenizer.get_decoder_prompt_ids(language='en',
+                                                                                         task="transcribe")
 
-options = dict(language='es', beam_size=5, best_of=5)
-transcribe_options = dict(task="transcribe", **options)
-translate_options = dict(task="translate", **options)
+translate = pipeline(
+    task="automatic-speech-recognition",
+    model=MODEL_NAME,
+    chunk_length_s=30,
+    device=device,
+)
+transcribe.model.config.forced_decoder_ids = transcribe.tokenizer.get_decoder_prompt_ids(language='es',
+                                                                                         task="translate")
 
 
 def transcribe(audio):
-    transcription = model.transcribe(audio, **transcribe_options)
-    translation = model.transcribe(audio, **translate_options)
+    transcription = transcribe(audio)
+    translation = translate(audio)
     return transcription["text"], translation["text"]