Float the preds

app.py

@@ -19,7 +19,7 @@ def _preprocess(image_path: str):
 def predict(image):
     image_array = _preprocess(image)
     predictions = model.predict(image_array)[0]  # single pred batch
-    return {k:v for k, v in zip(CLASS_LABELS, predictions)}
+    return {k: float(v) for k, v in zip(CLASS_LABELS, predictions)}
 
 gr.Interface(
     predict,