Update app.py

app.py

@@ -20,7 +20,10 @@ def classify_pet(image):
     clip_results = clip_detector(image, candidate_labels=labels_oxford_pets)
     clip_output = {result['label']: result['score'] for result in clip_results}
     
-    return {"ViT Classification": vit_output, "CLIP Zero-Shot Classification": clip_output}
+    return {
+        "ViT Classification": vit_output,
+        "CLIP Zero-Shot Classification": clip_output
+    }
 
 example_images = [
     ["example_images/dog1.jpeg"],
@@ -38,8 +41,10 @@ iface = gr.Interface(
     description="Upload an image of a pet, and compare results from a trained ViT model and a zero-shot CLIP model.",
     examples=[
         ["example_images/dog1.jpeg"],
-        ["example_images/cat.jpg"],
-        ["example_images/leonberger.jpg"]
+        ["example_images/dog2.jpeg"],
+        ["example_images/leonberger.jpg"],
+        ["example_images/snow_leopard.jpeg"],
+        ["example_images/cat.jpg"]
     ],
     cache_examples=False
 )