Update app.py

app.py

@@ -18,6 +18,7 @@ from transformers import (
 from transformers.pipelines.audio_utils import ffmpeg_read
 import torch  # If you're using PyTorch
 from datasets import load_dataset, Dataset, DatasetDict
+import spaces
 
 # Constants
 MODEL_NAME = "openai/whisper-large-v3"
@@ -101,6 +102,7 @@ def download_yt_audio(yt_url, filename):
         ydl.download([yt_url])
     return info
 
+@spaces.GPU(duration=120)
 def yt_transcribe(yt_url, task):
     # Load the dataset
     dataset = load_dataset(DATASET_NAME, split="train")