gelnesr commited on
Commit
f87a26a
·
verified ·
1 Parent(s): 6958775
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -140,7 +140,7 @@ def handle_name(name=None, pdb_input=None, model_version="ESM3"):
140
  return f'{pdb_name}-Dyna1{"" if model_version == "ESM3" else "-ESM2"}'
141
 
142
  @spaces.GPU(duration=300)
143
- def run_model(model, model_version='ESM2', seq_input=None, struct_input=None, sequence_id=None):
144
  if model_version == "ESM3":
145
  model = ESM_model(method='esm3')
146
  model.load_state_dict(torch.load('Dyna-1/model/weights/dyna1.pt'), strict=False)
 
140
  return f'{pdb_name}-Dyna1{"" if model_version == "ESM3" else "-ESM2"}'
141
 
142
  @spaces.GPU(duration=300)
143
+ def run_model(model_version='ESM2', seq_input=None, struct_input=None, sequence_id=None):
144
  if model_version == "ESM3":
145
  model = ESM_model(method='esm3')
146
  model.load_state_dict(torch.load('Dyna-1/model/weights/dyna1.pt'), strict=False)