reduce to 1200s
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 predict_dynamics(sequence=None, pdb_input=None, chain_id='A', use_pdb_seq=False, model_version="ESM3", name=None, oauth_token: Optional[str] = None):
|
| 144 |
try:
|
| 145 |
# Validate ESM2 requires sequence
|
|
|
|
| 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=1200)
|
| 143 |
def predict_dynamics(sequence=None, pdb_input=None, chain_id='A', use_pdb_seq=False, model_version="ESM3", name=None, oauth_token: Optional[str] = None):
|
| 144 |
try:
|
| 145 |
# Validate ESM2 requires sequence
|