Update app.py

app.py

@@ -35,7 +35,6 @@ class ModelManager:
         else:
             return load_model(model_path)
 
-
 class ImageProcessor:
 
     def process_image(self, image: Image.Image, model_name: str):
@@ -119,7 +118,6 @@ class ImageProcessor:
             img = cv2.cvtColor(img, cv2.COLOR_GRAY2RGB)
         return img
 
-
 class GradioInterface:
     def __init__(self):
         self.image_processor = ImageProcessor()
@@ -261,16 +259,6 @@ class GradioInterface:
         </div>
         """
 
-        header_html = f"""
-        <div class="app-header">
-            <h1 class="app-title">AI in Dentistry</h1>
-            <h2 class="app-subtitle"> Advancing Imaging and Clinical Transcription</h2>
-            <p class="app-description">
-                This application demonstrates the use of AI in dentistry for tasks such as classification, detection, and segmentation.
-            </p>
-        </div>
-        """
-
         js_func = """
         function refresh() {
             const url = new URL(window.location);
@@ -300,10 +288,21 @@ class GradioInterface:
                             choices=list(Config.MODELS.keys()),
                             value="Calculus and Caries Classification",
                         )
-                    examples = gr.Examples(
+                    examples_classification = gr.Examples(
+                        label="Classification Examples",
                         inputs=input_image,
                         examples=self.preloaded_examples["Calculus and Caries Classification"],
                     )
+                    examples_detection = gr.Examples(
+                        label="Caries Detection Examples",
+                        inputs=input_image,
+                        examples=self.preloaded_examples["Caries Detection"],
+                    )
+                    examples_segmentation = gr.Examples(
+                        label="Segmentation Examples",
+                        inputs=input_image,
+                        examples=self.preloaded_examples["Dental X-Ray Segmentation"],
+                    )
                 with gr.Column():
                     result = gr.Image(label="Result", elem_classes="image-preview")
                     run_button = gr.Button("Run", elem_classes="gr-button")
@@ -311,7 +310,7 @@ class GradioInterface:
             model_name.change(
                 fn=update_examples,
                 inputs=model_name,
-                outputs=examples.dataset,
+                outputs=[examples_classification.dataset, examples_detection.dataset, examples_segmentation.dataset],
             )
 
             run_button.click(
@@ -328,5 +327,4 @@ def main():
     demo.launch(share=False)
 
 if __name__ == "__main__":
-    main()
-
+    main()