Update app.py

app.py

@@ -6,6 +6,7 @@ import os
 import tempfile
 import torch
 from transformers import WhisperProcessor, WhisperForConditionalGeneration
+import spaces
 
 def extract_audio(video_path):
     video = mp.VideoFileClip(video_path)
@@ -14,6 +15,7 @@ def extract_audio(video_path):
     audio.write_audiofile(audio_path)
     return audio_path
 
+@spaces.GPU
 def generate_subtitles(audio_path):
     device = "cuda" if torch.cuda.is_available() else "cpu"
     processor = WhisperProcessor.from_pretrained("openai/whisper-large-v3")