Update app.py

app.py

@@ -63,7 +63,7 @@ def predict(text):
     'label': class_names[indices.cpu().numpy()[0]],
     'probablities': {class_names[i] : round(probs[0][i].item(),3) for i in range(len(probs[0]))}
   }
-  return data
+  return {class_names[i] : round(probs[0][i].item(),3) for i in range(len(probs[0]))}
 
 gr.Interface(
     predict,