Update app.py
Browse files
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=
|
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)
|