Simon Duerr commited on
Commit
a43f8f1
·
1 Parent(s): e1f203f

fix model weights path

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -189,7 +189,7 @@ def esm(protein_path, out_file):
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",
 
189
  # create args object with defaults
190
  os.environ["HOME"] = "esm/model_weights"
191
  print("calling ", sys.executable)
192
+ print("model weights", os.listdir("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",