streamlit doesn't handle floats well

language_models_project/app.py

@@ -71,7 +71,7 @@ def infer(text: str) -> List[Dict[str, float]]:
     predictions[np.where(probs >= 0.5)] = 1
     predictions = pd.Series(predictions == 1)
 
-    l = pd.Series(zip(predictions.tolist(), probs.tolist()))
+    l = pd.Series(zip(predictions.tolist(), probs.tolist())).apply(str)
     l.index = [
         "toxic",
         "severe_toxic",