Update app.py

app.py

@@ -13,13 +13,13 @@ all_label_names = list(model.config.id2label.values())
 
 def predict(text):
     probs = expit(model(**tokenizer([texts], return_tensors="pt", padding=True)).logits.detach().numpy())
-    return list(zip(all_label_names, probs[0]))
+    return {i: np.round(j, 2) for i, j in zip(all_label_names, probs[0])}
 
 iface = gr.Interface(
   fn=predict, 
-  inputs='What is going on with you',
-  outputs='Our predictions',
-  examples=[["This test tomorrow is really freaking me out."]]
+  inputs="text",
+  outputs="label",
+  examples=["This test tomorrow is really freaking me out."]
 )
 
 iface.launch()