Spaces:
Sleeping
Sleeping
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
|