Update app.py

Still need to fix -- "cuda:0"

app.py

@@ -82,7 +82,7 @@ fold_vocab = {
 def predict(model_name, pairs_file, sequence_file, progress = gr.Progress()):
 
     run_id = uuid4()
-    device = torch.device("0") if torch.cuda.is_available() else torch.device("cpu")
+    device = torch.device("cuda:0") if torch.cuda.is_available() else torch.device("cpu")
 
     # gr.Info("Loading model...")
     _ = lm_embed("M", use_cuda = (device.type == "cuda"))