gelnesr commited on
Commit
5ebff22
·
verified ·
1 Parent(s): 2d43f3d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -139,7 +139,7 @@ def handle_name(name=None, pdb_input=None, model_version="ESM3"):
139
  pdb_name = str(random.randint(0, 100000))
140
  return f'{pdb_name}-Dyna1{"" if model_version == "ESM3" else "-ESM2"}'
141
 
142
- @spaces.GPU(duration=20)
143
  def run_model(model, model_version='ESM2', seq_input=None, struct_input=None, sequence_id=None):
144
  if model_version == "ESM3":
145
  logits = model((seq_input, struct_input), sequence_id)
 
139
  pdb_name = str(random.randint(0, 100000))
140
  return f'{pdb_name}-Dyna1{"" if model_version == "ESM3" else "-ESM2"}'
141
 
142
+ @spaces.GPU(duration=50)
143
  def run_model(model, model_version='ESM2', seq_input=None, struct_input=None, sequence_id=None):
144
  if model_version == "ESM3":
145
  logits = model((seq_input, struct_input), sequence_id)