wwydmanski commited on
Commit
137a7d5
1 Parent(s): c86f7b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ def get_esmfold_embeddings(sequence):
45
  with torch.no_grad():
46
  output = model(tokenized_input)
47
 
48
- return {"res": output["s_s"].cpu().numpy()}
49
 
50
  def suggest(option):
51
  if option == "Plastic degradation protein":
 
45
  with torch.no_grad():
46
  output = model(tokenized_input)
47
 
48
+ return {"res": output["s_s"].cpu().tolist()}
49
 
50
  def suggest(option):
51
  if option == "Plastic degradation protein":