pass model through
Browse files
app.py
CHANGED
@@ -193,7 +193,7 @@ def predict_dynamics(sequence=None, pdb_input=None, chain_id='A', use_pdb_seq=Fa
|
|
193 |
if not (sequence or (pdb_input and model_version == "ESM3")):
|
194 |
raise ValueError('Please provide a sequence' + (' or structure input' if model_version == "ESM3" else ''))
|
195 |
|
196 |
-
logits = run_model(model_version, seq_input, struct_input, sequence_id)
|
197 |
|
198 |
probabilities = utils.prob_adjusted(logits).cpu().detach().numpy()
|
199 |
|
|
|
193 |
if not (sequence or (pdb_input and model_version == "ESM3")):
|
194 |
raise ValueError('Please provide a sequence' + (' or structure input' if model_version == "ESM3" else ''))
|
195 |
|
196 |
+
logits = run_model(model, model_version, seq_input, struct_input, sequence_id)
|
197 |
|
198 |
probabilities = utils.prob_adjusted(logits).cpu().detach().numpy()
|
199 |
|