Update app.py

app.py

@@ -79,21 +79,16 @@ def predict(input_img):
 def create_bar_plot(predictions):
     data = pd.DataFrame(list(predictions.items()), columns=["Location", "Probability"])
     max_prob = data["Probability"].max()
-    
-    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
+    return gr.BarPlot(
+        data,
+        x="Location",
+        y="Probability",
+        title="Top 10 Predictions with Probabilities",
+        tooltip=["Location", "Probability"],
+        y_lim=[0, max_prob],
+        width=350,  # Set the width of the plot
+        height=600  # Set the height of the plot
+    )
 
 def predict_and_plot(input_img):
     predictions = predict(input_img)