Update app.py

app.py

@@ -7,7 +7,6 @@ import os
 import numpy as np
 
 
-# labels= {'Burger King': 0, 'KFC': 1,'McDonalds': 2,'Other': 3,'Starbucks': 4,'Subway': 5}
 HEIGHT,WIDTH=224,224
 IMG_SIZE=224
 model=load_model('Models/best_model1.h5')
@@ -51,15 +50,21 @@ def classify_image(inp):
     prediction = model.predict(inp).flatten()
     return {labels[i]: f"{prediction[i]:.6f}" for i in range(NUM_CLASSES)}
 
-# image = gr.inputs.Image(shape=(IMG_SIZE, IMG_SIZE))
-# label = gr.outputs.Label(num_top_classes=2)
-
-# gr.Interface(fn=classify_image, inputs=image, outputs=label, title='Cats Vs Dogs',height=600, width=1200,examples=ex,theme='peach').launch(debug=True)
-
-    
 image = gr.Image(height=HEIGHT,width=WIDTH,label='Input')
 label = gr.Label(num_top_classes=2)
 
-gr.Interface(fn=classify_image, inputs=image, outputs=label, title='Smart Pet Classifier').launch(debug=False)
+examples = [
+    ["Examples/img1.png"],
+    ["Examples/img2.png"],
+    ["Examples/img3.png"],
+    ["Examples/img4.png"]    
+]
 
 
+gr.Interface(
+    fn=classify_image, 
+    inputs=image, 
+    outputs=label, 
+    title='Smart Pet Classifier',
+    examples=examples
+).launch(debug=False)