Update app.py

app.py

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