Sukanyaaa commited on
Commit
605fad6
·
1 Parent(s): 06baf9d

fix inference_app.py

Browse files
Files changed (1) hide show
  1. inference_app.py +1 -1
inference_app.py CHANGED
@@ -888,7 +888,7 @@ model.eval()
888
  def predict (input_seq_1, input_msa_1, input_protein_1, input_seq_2,input_msa_2, input_protein_2):
889
  start_time = time.time()
890
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
891
- data = create_graph(input_protein_1, input_protein_2, '/home/sukanya/iitm_bisect_pinder_submission/test_out (1).pdb', k=10)
892
 
893
  with torch.no_grad():
894
  mat, vect = model(data)
 
888
  def predict (input_seq_1, input_msa_1, input_protein_1, input_seq_2,input_msa_2, input_protein_2):
889
  start_time = time.time()
890
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
891
+ data = create_graph(input_protein_1, input_protein_2, './test_out (1).pdb', k=10)
892
 
893
  with torch.no_grad():
894
  mat, vect = model(data)