Update app.py

app.py

@@ -1,6 +1,7 @@
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 from scipy.special import expit
 import numpy as np
+import os
 
 # set up model
 auth_token = os.environ.get("TOKEN") or True
@@ -9,9 +10,9 @@ model = AutoModelForSequenceClassification.from_pretrained("guidecare/feelings_a
 all_label_names = list(model.config.id2label.values())
 
 
-def probs(texts):
-    probs = expit(model(**tokenizer(texts, return_tensors="pt", padding=True)).logits.detach().numpy())
-    return list(zip(all_label_names, probs))
+def probs(text):
+    probs = expit(model(**tokenizer([texts], return_tensors="pt", padding=True)).logits.detach().numpy())
+    return list(zip(all_label_names, probs[0]))
 
 iface = gr.Interface(
   fn=predict,