Simon Duerr commited on
Commit
e1f203f
1 Parent(s): 16c9bf9
Files changed (2) hide show
  1. app.py +6 -1
  2. 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