Update app.py

app.py

@@ -6,8 +6,7 @@ import whisper
 from transformers import pipeline
 
 # Load the tiny Whisper model
-whisper_model = whisper.load_model("tiny")
-model = whisper.load_model("base")
+
 
 # Check if GPU is available and set device accordingly
 device = 0 if torch.cuda.is_available() else -1
@@ -16,6 +15,10 @@ if device == 0:
 else:
     print("Running on CPU")
     
+whisper_model = whisper.load_model("tiny", device=device)
+#model = whisper.load_model("base")
+
+
 # Load the text summarization model from Hugging Face
 summarizer = pipeline(task="summarization", model="facebook/bart-large-cnn", device=device)