simonduerr commited on
Commit
298284a
1 Parent(s): 2ef4a92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -191,9 +191,12 @@ def esm(protein_path, out_file):
191
  os.environ["HOME"] = "esm/model_weights"
192
  print("calling ", sys.executable)
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,
 
197
  )
198
 
199
 
 
191
  os.environ["HOME"] = "esm/model_weights"
192
  print("calling ", sys.executable)
193
  print("datadir", os.listdir("data"))
194
+ print(os.environ)
195
+ print(torch.____file__)
196
  subprocess.call(
197
  f"python esm/scripts/extract.py esm2_t33_650M_UR50D {out_file} data/esm2_output --repr_layers 33 --include per_tok",
198
  shell=True,
199
+ env=os.environ
200
  )
201
 
202