Update app.py

app.py

@@ -42,18 +42,11 @@ def create_pipe(model, flash):
     )
     return pipe, model  # Return both pipe and model for later GPU switch
 
-@spaces.GPU
+@spaces.GPU(duration=120)
 def move_to_gpu(model):
-    if torch.cuda.is_available():
-        device = "cuda:0"
-        torch_dtype = torch.float16  # Use float16 precision on GPU
-        model.to(device, dtype=torch_dtype)
-    elif platform == "darwin":
-        device = "mps"
-        model.to(device)
-    else:
-        device = "cpu"
-    
+    device = "cuda:0"
+    torch_dtype = torch.float16  # Use float16 precision on GPU
+    model.to(device, dtype=torch_dtype)    
     return device
 
 def transcribe_webui_simple_progress(modelName, languageName, urlData, multipleFiles, microphoneData, task, flash,