Update app.py

app.py

@@ -9,8 +9,15 @@ from transformers import pipeline
 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
+if device == 0:
+    print("Running on GPU")
+else:
+    print("Running on CPU")
+    
 # Load the text summarization model from Hugging Face
-summarizer = pipeline(task="summarization", model="facebook/bart-large-cnn")
+summarizer = pipeline(task="summarization", model="facebook/bart-large-cnn", device=device)
 
 # Function to transcribe and summarize the audio file
 def transcribe_and_summarize(audio):