Update app.py

app.py

@@ -36,12 +36,12 @@ MODEL_NAME = "openai/whisper-medium"
 FILE_LIMIT_MB = 1000
 YT_LENGTH_LIMIT_S = 3600
 
-device = 0 if torch.cuda.is_available() else "cpu"
+device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
 pipe = pipeline(
     task="automatic-speech-recognition",
     model=MODEL_NAME,
-    #chunk_length_s=30,
+    chunk_length_s=30,
     device=device,
     model_kwargs={"low_cpu_mem_usage": True},
 )