Update app.py

app.py

@@ -1,8 +1,11 @@
 import gradio as gr
 import pandas as pd
 import joblib
+import json
+from json import encoder
 
-
+encoder.FLOAT_REPR = lambda o: format(o, '.2f')
+ 
 model = joblib.load("nb_model.pickle")
 
 '''
@@ -21,7 +24,7 @@ def predict(text):
     #prediction_prob = round(model.predict_proba(data).max(), 3)
     #return prediction_class, prediction_prob
     pred = model.predict_proba(data)[0]
-    return {'Android': int(pred[0]), 'iPhone': int(pred[1])}
+    return {'Android': json.dumps(pred[0]), 'iPhone': json.dumps(pred[1])}
 
 
 description = "According to the dataset used for this model, Trump  mainly uses two devices for tweeting" \