Update app.py

app.py

@@ -4,7 +4,7 @@ import pytubefix as pt
 import os, time, librosa, torch
 from pyannote.audio import Pipeline
 from transformers import pipeline
-from google.colab import userdata
+import spaces
 
 
 def second_to_timecode(x: float) -> str:
@@ -60,7 +60,7 @@ asr_pipeline = pipeline(
     tokenizer=processor.tokenizer,
     feature_extractor=processor.feature_extractor,
     chunk_length_s=chunk_length_s, 
-    device=device
+    device_map="auto"
 )
 
 lang = 'mn'
@@ -68,6 +68,8 @@ asr_pipeline.model.config.forced_decoder_ids = asr_pipeline.tokenizer.get_decode
 
 print("----------> Loaded models <-----------")
 
+gpu_timeout = int(os.getenv("GPU_TIMEOUT", 60))
+@spaces.GPU(duration=gpu_timeout)
 def generator(youtube_link, microphone, file_upload, num_speakers, max_duration, history):
 
     if int(youtube_link != '') + int(microphone is not None) + int(file_upload is not None) != 1: