Update app.py

app.py

@@ -78,13 +78,14 @@ 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, 1],
+        y_lim=[0, max_prob],
         width=800,  # Set the width of the plot
         height=600  # Set the height of the plot
     )