Update app.py

app.py

@@ -13,7 +13,8 @@ all_label_names = list(model.config.id2label.values())
 
 def predict(text):
     probs = expit(model(**tokenizer([text], return_tensors="pt", padding=True)).logits.detach().numpy())
-    return {i: np.round(j, 2) for i, j in zip(all_label_names, probs[0])}
+    print(probs.shape, probs[0].shape)
+    return {i: np.round(j, 2) for i, j in zip(all_label_names, probs)}
 
 iface = gr.Interface(
   fn=predict,