Update app.py

app.py

@@ -1,6 +1,8 @@
 import os
 from huggingface_hub import Repository
 import numpy as np
+import pandas as pd
+
 
 # Retrieve the token from the environment variables
 token = os.environ.get("token")
@@ -74,10 +76,26 @@ def predict(input_img):
         results = {top_10_predictions[i]: float(top_10_probabilities[i]) for i in range(10)}
         return results
 
+def create_bar_plot(predictions):
+    data = pd.DataFrame(list(predictions.items()), columns=["Location", "Probability"])
+    return gr.BarPlot(
+        data,
+        x="Location",
+        y="Probability",
+        title="Top 10 Predictions with Probabilities",
+        tooltip=["Location", "Probability"],
+        y_lim=[0, 1]
+    )
+
+def predict_and_plot(input_img):
+    predictions = predict(input_img)
+    return create_bar_plot(predictions)
+
+    
 gradio_app = gr.Interface(
-    fn=predict,
+    fn=predict_and_plot,
     inputs=gr.Image(label="Upload an Image", type="pil"),
-    outputs=gr.JSON(label="Top 10 Predictions with Probabilities"),
+    outputs=gr.BarPlot(),
     title="Predict the Location of this Image"
 )