Added progres

app.py

@@ -41,14 +41,14 @@ def aa_embed(sequence: str):
     tokens = tokenizer_aa([sequence], return_tensors="pt")
     with torch.no_grad():
         torch_outs = model_aa(**tokens)
-    return torch_outs
+    return torch_outs[0]
 
 
 def se_embed(sentence: str):
     encoded_input = tokenizer_se([sentence], return_tensors='pt')
     with torch.no_grad():
         model_output = model_se(**encoded_input)
-    return model_output
+    return model_output[0]
 
 
 def download_data_if_required():

requirements.txt

@@ -8,4 +8,5 @@ torch-geometric==2.3.1
 --find-links https://data.pyg.org/whl/torch-2.0.0+cpu.html torch-scatter==2.1.1
 --find-links https://data.pyg.org/whl/torch-2.0.0+cpu.html torch-sparse==0.6.17
 --find-links https://data.pyg.org/whl/torch-2.0.0+cpu.html torch-spline-conv==1.2.2
-transformers==4.29.2
+transformers==4.29.2
+progres