Update app.py

app.py

@@ -2,12 +2,16 @@ import gradio as gr
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 
 def processInput(input): 
+    #load model and tokenizer
     model = AutoModelForSequenceClassification.from_pretrained("garrettbaber/twitter-roberta-base-fear-intensity")
     tokenizer = AutoTokenizer.from_pretrained("garrettbaber/twitter-roberta-base-fear-intensity")
+    #get tokens
     tokens = tokenizer(input, return_tensors="pt")
-    print('Running tokens in model')
+    #pass tokens to model
     outputs = model(**tokens)
-    return outputs
+    #parse output
+    logits = outputs.get("logits")
+    return logits.tolist()
 
 app = gr.Interface(fn=processInput, inputs="text", outputs="text")