Update app.py

app.py

@@ -141,8 +141,9 @@ def main(file_name):
         
     hf_token = os.getenv("HUGGINGFACE_TOKEN")
     pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1",use_auth_token=hf_token)
-    device = torch.device('cuda:0') if torch.cuda.is_available() else "cpu")#torch.device('cpu')
+    #device = torch.device('cuda:0') if torch.cuda.is_available() else "cpu"#torch.device('cpu')
     #device = torch.device('cpu')
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
     video_file = file_name #deepfake #meganfox.mp4'
     #current_path = os.getcwd()