.numpy()

app.py

@@ -145,7 +145,7 @@ def run_model(model, model_version='ESM2', seq_input=None, struct_input=None, se
         logits = model((seq_input, struct_input), sequence_id)
     else:
         logits = model(seq_input, sequence_id)
-    return logits.cpu().detach().numpy()
+    return logits.cpu().detach()
 
 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):
     try:
@@ -195,7 +195,7 @@ def predict_dynamics(sequence=None, pdb_input=None, chain_id='A', use_pdb_seq=Fa
         
         logits = run_model(model, model_version, seq_input, struct_input, sequence_id)
         
-        probabilities = utils.prob_adjusted(logits)
+        probabilities = utils.prob_adjusted(logits).numpy()
         
         seq_to_use = sequence if sequence else pdb_seq if pdb_input else sequence
         results_df = pd.DataFrame({