Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 from huggingface_hub import Repository
 import numpy as np
 import pandas as pd
-
+import matplotlib.pyplot as plt
 
 # Retrieve the token from the environment variables
 token = os.environ.get("token")
@@ -79,21 +79,26 @@ def predict(input_img):
 def create_bar_plot(predictions):
     data = pd.DataFrame(list(predictions.items()), columns=["Location", "Probability"])
     max_prob = data["Probability"].max()
-    return gr.BarPlot(
-        data,
-        x="Location",
-        y="Probability",
-        title="Top 10 Predictions with Probabilities",
-        tooltip=["Location", "Probability"],
-        y_lim=[0, max_prob],
-        width=400,  # Set the width of the plot
-        height=300  # Set the height of the plot
-    )
+    
+    fig, ax = plt.subplots(figsize=(10, 6))
+    ax.barh(data["Location"], data["Probability"], color='skyblue')
+    ax.set_xlabel('Probability', fontsize=14)
+    ax.set_title('Top 10 Predictions with Probabilities', fontsize=16)
+    ax.set_xlim(0, max_prob)
+    ax.tick_params(axis='y', labelsize=12)  # Increase font size for y-axis labels
+    ax.invert_yaxis()
+    
+    # Save the plot to a BytesIO object
+    buf = io.BytesIO()
+    plt.savefig(buf, format='png')
+    buf.seek(0)
+    
+    return buf
 
 def predict_and_plot(input_img):
     predictions = predict(input_img)
     return create_bar_plot(predictions)
-
+    
     
 gradio_app = gr.Interface(
     fn=predict_and_plot,