Spaces:

savakholin
/

esm-2

Paused

savakholin commited on Jul 7, 2023

Commit

ced1852

1 Parent(s): a931815

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import json
 def embed(aa_seq, tokenizer, model):
     inputs = tokenizer(aa_seq, return_tensors="pt")
     outputs = model(**inputs)
-    last_hidden_states = outputs.last_hidden_state.to_list()
     return last_hidden_states

 def embed(aa_seq, tokenizer, model):
     inputs = tokenizer(aa_seq, return_tensors="pt")
     outputs = model(**inputs)
+    last_hidden_states = outputs.last_hidden_state.detach().numpy().tolist()
     return last_hidden_states