back to previous data load

run_prothgt_app.py

@@ -135,7 +135,7 @@ def generate_prediction_df(protein_ids, model_paths, model_config_paths, go_cate
         print(f"Error downloading file: {e}")
         raise
 
-    heterodata = torch.load(output, weights_only=False)
+    heterodata = torch.load(output)
     print(heterodata.edge_types)
     
     # Remove unnecessary edge types