Update app.py

app.py

@@ -162,9 +162,10 @@ def compute_plddt_iptm(protein_seq, binder_seq):
 
     # based on https://colab.research.google.com/github/huggingface/notebooks/blob/main/examples/protein_folding.ipynb 
     model = EsmForProteinFolding.from_pretrained("facebook/esmfold_v1", low_cpu_mem_usage=True)
-    model = model.cuda()
+    #model = model.cuda()
     model.esm = model.esm.half()     # Uncomment to switch the stem to float16
-    torch.backends.cuda.matmul.allow_tf32 = True
+    #torch.backends.cuda.matmul.allow_tf32 = True
+    print("168:model.device in compute_plddt_iptm",model.device)
     
     sequence = protein_seq + binder_seq
     inputs = tokenizer(sequence, return_tensors='pt', add_special_tokens=False)