Update app.py

app.py

@@ -45,7 +45,7 @@ def get_esmfold_embeddings(sequence):
     with torch.no_grad():
         output = model(tokenized_input)
 
-    return output["s_s"].cpu().numpy()
+    return {"res": output["s_s"].cpu().numpy()}
 
 def suggest(option):
    if option == "Plastic degradation protein":