Spaces:
Running
Running
Erva Ulusoy
commited on
Commit
·
733cb6a
1
Parent(s):
8aa6c67
possible fix for pt file load
Browse files- 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)
|
| 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, weights_only=False)
|
| 139 |
print(heterodata.edge_types)
|
| 140 |
|
| 141 |
# Remove unnecessary edge types
|