Erva Ulusoy commited on
Commit
83d87f5
·
1 Parent(s): e3a9f09

back to previous data load

Browse files
Files changed (1) hide show
  1. run_prothgt_app.py +1 -1
run_prothgt_app.py CHANGED
@@ -135,7 +135,7 @@ def generate_prediction_df(protein_ids, model_paths, model_config_paths, go_cate
135
  print(f"Error downloading file: {e}")
136
  raise
137
 
138
- heterodata = torch.load(output, weights_only=False)
139
  print(heterodata.edge_types)
140
 
141
  # Remove unnecessary edge types
 
135
  print(f"Error downloading file: {e}")
136
  raise
137
 
138
+ heterodata = torch.load(output)
139
  print(heterodata.edge_types)
140
 
141
  # Remove unnecessary edge types