Update app.py

app.py

@@ -47,6 +47,7 @@ def classify_image(pil_image):
     plt.ylabel('Probability')
     plt.title('Prediction Probabilities')
     plt.savefig('prediction_plot.png')
+    plt.close()
 
     
     output_text = f"Predicted Class: {predicted_class_name}, Confidence: {confidence:.4f}\n"
@@ -56,7 +57,16 @@ def classify_image(pil_image):
     return output_text, 'prediction_plot.png'
 
 
-iface = gr.Interface(fn=classify_image, inputs="image", outputs=["text", "image"],live=True)
+iface = gr.Interface(fn=classify_image, 
+                     inputs="image", 
+                     outputs=["text", "image"],
+                     examples=[
+                         "L9_1b95a3808073c0edad3454d1dedf3dcc.jpg",
+                         "L6_0a171beb21a6f4d6fef31f8ccb400eae.jpg",
+                         "L2_00a6b5e9806a8b072b98fdeacb3f45b5.jpg",
+                         "L4_0b02898e9d31954dd5378e0ffbdb9a41.jpg"
+                     ],
+                     live=True)
 
 
 iface.launch()