Update app.py

app.py

@@ -1,4 +1,3 @@
-import accelerate
 import gradio as gr
 import time
 import io
@@ -12,7 +11,7 @@ from transformers import AutoModelForSpeechSeq2Seq, AutoProcessor, pipeline
 
 model = AutoModelForSpeechSeq2Seq.from_pretrained(pretrained_model_name_or_path= "openai/whisper-large-v3",
                                                   torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
-                                                  use_safetensors=True).to("cuda")
+                                                  use_safetensors=True)
 
 #Instantiating the processor object.
 
@@ -28,8 +27,8 @@ pipe = pipeline(task="automatic-speech-recognition",
                 chunk_length_s=30,
                 batch_size=16,
                 return_timestamps=True,
-                torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
-                device="cuda")
+                torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
+               )
 
 #Defining speech-to-text function.