Spaces:
Runtime error
Runtime error
Simon Duerr
commited on
Commit
·
e1f203f
1
Parent(s):
16c9bf9
add debug
Browse files- app.py +6 -1
- esm/scripts/extract.py +2 -1
app.py
CHANGED
|
@@ -181,11 +181,16 @@ def molecule(input_pdb, ligand_pdb):
|
|
| 181 |
allowpaymentrequest="" frameborder="0" srcdoc='{x}'></iframe>"""
|
| 182 |
|
| 183 |
|
|
|
|
|
|
|
|
|
|
| 184 |
def esm(protein_path, out_file):
|
| 185 |
esm_embedding_prep(out_file, protein_path)
|
| 186 |
# create args object with defaults
|
| 187 |
os.environ["HOME"] = "esm/model_weights"
|
| 188 |
-
|
|
|
|
|
|
|
| 189 |
subprocess.call(
|
| 190 |
f"python esm/scripts/extract.py esm2_t33_650M_UR50D {out_file} data/esm2_output --repr_layers 33 --include per_tok",
|
| 191 |
shell=True,
|
|
|
|
| 181 |
allowpaymentrequest="" frameborder="0" srcdoc='{x}'></iframe>"""
|
| 182 |
|
| 183 |
|
| 184 |
+
import sys
|
| 185 |
+
|
| 186 |
+
|
| 187 |
def esm(protein_path, out_file):
|
| 188 |
esm_embedding_prep(out_file, protein_path)
|
| 189 |
# create args object with defaults
|
| 190 |
os.environ["HOME"] = "esm/model_weights"
|
| 191 |
+
print("calling ", sys.executable)
|
| 192 |
+
print("model weights", os.listdir("esm/model_weights"))
|
| 193 |
+
print("datadir", os.listdir("data"))
|
| 194 |
subprocess.call(
|
| 195 |
f"python esm/scripts/extract.py esm2_t33_650M_UR50D {out_file} data/esm2_output --repr_layers 33 --include per_tok",
|
| 196 |
shell=True,
|
esm/scripts/extract.py
CHANGED
|
@@ -6,7 +6,8 @@
|
|
| 6 |
|
| 7 |
import argparse
|
| 8 |
import pathlib
|
| 9 |
-
|
|
|
|
| 10 |
import torch
|
| 11 |
|
| 12 |
from esm import Alphabet, FastaBatchedDataset, ProteinBertModel, pretrained, MSATransformer
|
|
|
|
| 6 |
|
| 7 |
import argparse
|
| 8 |
import pathlib
|
| 9 |
+
import sys
|
| 10 |
+
print("using", sys.executable)
|
| 11 |
import torch
|
| 12 |
|
| 13 |
from esm import Alphabet, FastaBatchedDataset, ProteinBertModel, pretrained, MSATransformer
|