Update app.py

app.py

@@ -5,6 +5,7 @@ from googletrans import Translator
 from pydub import AudioSegment
 import os
 import tempfile
+import spaces
 
 def extract_audio(video_path):
     video = mp.VideoFileClip(video_path)
@@ -13,6 +14,7 @@ def extract_audio(video_path):
     audio.write_audiofile(audio_path)
     return audio_path
 
+@spaces.GPU
 def generate_subtitles(audio_path):
     model = whisper.load_model("base")
     result = model.transcribe(audio_path)