Update app.py

app.py

@@ -53,15 +53,43 @@ class Segmenter:
         output_predictions: torch.Tensor = output.argmax(0)
         return self.colorizer.colorize(output_predictions)
 
-segmenter = Segmenter()
+class GradioApp:
+    def __init__(self, segmenter: Segmenter):
+        self.segmenter = segmenter
 
-interface = gr.Interface(
-    fn=segmenter.segment,
-    inputs=gr.Image(type="pil"),
-    outputs=gr.Image(type="pil"),
-    title="Deeplabv3 Segmentation",
-    description="Upload an image to perform semantic segmentation using Deeplabv3 ResNet101."
-)
+    def launch(self):
+        with gr.Blocks() as demo:
+            gr.Markdown("<h1 style='text-align: center; color: #4CAF50;'>Deeplabv3 Segmentation</h1>")
+            gr.Markdown("<p style='text-align: center;'>Upload an image to perform semantic segmentation using Deeplabv3 ResNet101.</p>")
+            gr.Markdown("""
+                ### Model Information
+                **DeepLabv3 with ResNet101** is a convolutional neural network model designed for semantic image segmentation. 
+                It utilizes atrous convolution to capture multi-scale context by using different atrous rates.
+                """)
+            with gr.Row():
+                with gr.Column():
+                    image_input = gr.Image(type='pil', label="Input Image", show_label=False)
+                with gr.Column():
+                    image_output = gr.Image(type='pil', label="Segmented Output", show_label=False)
+                    
+            button = gr.Button("Segment")
+            button.click(fn=self.segmenter.segment, inputs=image_input, outputs=image_output)
+
+            gr.Markdown("### Example Images")
+            gr.Examples(
+                examples=[
+                    ["https://www.timeforkids.com/wp-content/uploads/2024/01/Snapshot_20240126.jpg?w=1024"],
+                    ["https://www.timeforkids.com/wp-content/uploads/2023/09/G3G5_230915_puffins_on_the_rise.jpg?w=1024"],
+                    ["https://www.timeforkids.com/wp-content/uploads/2024/03/G3G5_240412_bug_eyed.jpg?w=1024"]
+                ],
+                inputs=image_input,
+                outputs=image_output,
+                label="Click an example to use it"
+            )
+            
+        demo.launch()
 
 if __name__ == "__main__":
-    interface.launch()
+    segmenter = Segmenter()
+    app = GradioApp(segmenter)
+    app.launch()