added example images

app.py

@@ -1,12 +1,15 @@
 import gradio as gr
 from ultralytics import YOLO
+import os
 
+# catgories
 format = { 0: 'Bengin case',
              1: 'Bengin case Malignant case',
              2: 'Malignant case',
              3: 'Malignant case Normal case',
              4: 'Normal case'}
 
+# returning classifiers output
 def image_classifier(inp):
     model = YOLO("best.pt")
 
@@ -14,10 +17,25 @@ def image_classifier(inp):
     probs = result[0].probs
     max_tensor = max(probs)
     tensor_pos = ((probs == max_tensor).nonzero(as_tuple=True)[0])
-    
+
     return format.get(int(tensor_pos))
 
-web = gr.Interface(fn=image_classifier, inputs="image", outputs="text")
+# gradio code block for input and output
+with gr.Blocks() as app:
+    gr.Markdown("## Lung Cancer classification using Yolov8")
+    with gr.Row():
+        inp_img = gr.Image()
+        out_txt = gr.Textbox()
+    btn = gr.Button(value="Submit")
+    btn.click(image_classifier, inputs=inp_img, outputs=out_txt)
 
-web.launch()
+    gr.Markdown("## Image Examples")
+    gr.Examples(
+        examples=[os.path.join(os.path.dirname(__file__), "1.jpg"), os.path.join(os.path.dirname(__file__), "2.jpg")],
+        inputs=inp_img,
+        outputs=out_txt,
+        fn=image_classifier,
+        cache_examples=True,
+    )
 
+app.launch(share=True)