update app.py

app.py

@@ -17,6 +17,42 @@ MODELS = {
     "Tire Cord": "smartfalcon-ai/Tire-Cord-Defect-Detection"
 }
 
+# Example images for each model (3 examples per task)
+EXAMPLES = [
+    # Dental Implant
+    ["examples/dental-implant-1.jpg", "Dental Implant", 0.25],
+    ["examples/dental-implant-2.jpg", "Dental Implant", 0.25],
+    ["examples/dental-implant-3.jpg", "Dental Implant", 0.25],
+    # Data Matrix
+    ["examples/data-matrix-1.jpg", "Data Matrix", 0.25],
+    ["examples/data-matrix-2.jpg", "Data Matrix", 0.25],
+    ["examples/data-matrix-3.jpg", "Data Matrix", 0.25],
+    # Ball Pen
+    ["examples/ball-pen-1.jpg", "Ball Pen", 0.25],
+    ["examples/ball-pen-2.jpg", "Ball Pen", 0.25],
+    ["examples/ball-pen-3.jpg", "Ball Pen", 0.25],
+    # Knit Up
+    ["examples/knit-up-1.jpg", "Knit Up", 0.25],
+    ["examples/knit-up-2.jpg", "Knit Up", 0.25],
+    ["examples/knit-up-3.jpg", "Knit Up", 0.25],
+    # Knit Back
+    ["examples/knit-back-1.jpg", "Knit Back", 0.25],
+    ["examples/knit-back-2.jpg", "Knit Back", 0.25],
+    ["examples/knit-back-3.jpg", "Knit Back", 0.25],
+    # Jean Back
+    ["examples/jean-back-1.jpg", "Jean Back", 0.25],
+    ["examples/jean-back-2.jpg", "Jean Back", 0.25],
+    ["examples/jean-back-3.jpg", "Jean Back", 0.25],
+    # Jean Up
+    ["examples/jean-up-1.jpg", "Jean Up", 0.25],
+    ["examples/jean-up-2.jpg", "Jean Up", 0.25],
+    ["examples/jean-up-3.jpg", "Jean Up", 0.25],
+    # Tire Cord
+    ["examples/tire-cord-1.jpg", "Tire Cord", 0.25],
+    ["examples/tire-cord-2.jpg", "Tire Cord", 0.25],
+    ["examples/tire-cord-3.jpg", "Tire Cord", 0.25],
+]
+
 sessions = {}
 
 def get_session(model_name):
@@ -108,7 +144,8 @@ demo = gr.Interface(
     ],
     outputs=gr.Image(type="numpy"),
     title="Industrial Defect Detection",
-    description="Upload an image, select the defect model, and adjust the confidence threshold."
+    description="Upload an image, select the defect model, and adjust the confidence threshold.",
+    examples=EXAMPLES
 )
 
 if __name__ == "__main__":