aopp

app.py

@@ -19,15 +19,12 @@ def predict_text(text):
     predicted_class_index = prediction['class_index']
     predicted_class_label = list(class_label_mapping.keys())[list(class_label_mapping.values()).index(predicted_class_index)]
     predicted_class_probability = prediction['probs'][predicted_class_index]
-    return {"Label": predicted_class_label, "Probability": predicted_class_probability}
+    return {predicted_class_label: float(predicted_class_probability)}
 
 iface = gr.Interface(
     fn=predict_text,
     inputs="text",
-    outputs=[
-        gr.outputs.Textbox(label="Label"),
-        gr.outputs.Label(num_top_classes=1, type="confidences", label="Probability")
-    ],
+    outputs=gr.outputs.KeyValues(label="Predicted Class and Probability"),
     title="Food Origin Classification App",
     description="Enter a Recipe, and it will predict the class label.",
 )