lol typo
Browse files
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(
|
| 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)
|