Update app.py

app.py

@@ -36,6 +36,7 @@ def download_youtube_audio(url, output_path, preferred_quality="192"):
         return None
 
 # Function to transcribe audio using WhisperModel
+@spaces.GPU(duration=120)
 def transcribe(path, model_name):
     model = WhisperModel(model_name)
     print(f"Reading {path}")
@@ -43,6 +44,7 @@ def transcribe(path, model_name):
     return segments
 
 # Function to process segments and convert them into a DataFrame
+@spaces.GPU(duration=120)
 def process_segments(segments):
     result = {}
     print("Processing...")
@@ -60,6 +62,7 @@ def process_segments(segments):
     return df
 
 # Gradio interface functions
+@spaces.GPU(duration=120)
 def generate_transcript(youtube_url, model_name="distil-large-v3"):
     path = "downloaded_audio.mp3"
     download_youtube_audio(youtube_url, path)