Update app.py

app.py

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