gelnesr commited on
Commit
34a38bb
·
verified ·
1 Parent(s): 16f71d3

pass model through

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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