Format Output

app.py

@@ -1,6 +1,12 @@
 import gradio as gr
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 
+def formatOutput(modelOutput):
+    #parse output
+    logits = outputs.get("logits")
+    rawScore = logits.tolist().pop().pop()
+    return f"{rawScore:.3f}"
+
 def processInput(input): 
     #load model and tokenizer
     model = AutoModelForSequenceClassification.from_pretrained("garrettbaber/twitter-roberta-base-fear-intensity")
@@ -9,9 +15,7 @@ def processInput(input):
     tokens = tokenizer(input, return_tensors="pt")
     #pass tokens to model
     outputs = model(**tokens)
-    #parse output
-    logits = outputs.get("logits")
-    return logits.tolist()
+    return formatOutput(outputs)
 
 app = gr.Interface(fn=processInput, inputs="text", outputs="text")