Spaces:
Sleeping
Sleeping
fix inference_app.py
Browse files- 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, '
|
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)
|