Update app.py

app.py

@@ -3,8 +3,8 @@ import gradio as gr
 from TheDistanceAssessor import run, load_segformer, load_yolo
 
 def process_image(input_image):
-    image_size = [1024,1024]
-    target_distances = [650,1000,2000]
+    image_size = [1024, 1024]
+    target_distances = [650, 1000, 2000]
     num_ys = 10
 
     PATH_model_seg = 'SegFormer_B3_1024_finetuned.pth'
@@ -13,7 +13,7 @@ def process_image(input_image):
     model_det = load_yolo(PATH_model_det)
     
     input_image = cv2.cvtColor(input_image, cv2.COLOR_BGR2RGB)
-    output_image = run(input_image, model_seg, model_det, image_size, target_distances, num_ys = num_ys)
+    output_image = run(input_image, model_seg, model_det, image_size, target_distances, num_ys=num_ys)
     return output_image
 
 # Create the Gradio interface
@@ -22,9 +22,28 @@ iface = gr.Interface(
     inputs=gr.Image(type="numpy"),  # Input type
     outputs=gr.Image(type="numpy"),  # Output type
     title="RailSafeNet - Automatic Detection of Objects in the Track",  # Title of the interface
-    description="This is a demo of the master's thesis focused on the Automatic Detection of Objects in the Track.\n The repository with the code is accesible from: https://github.com/oValach/RailSafeNet_DT \n\nUpload an image with a scene including rail track and get a processed image with marked rail critical areas and detected and classified objects."
+    description="This is a demo of the master's thesis focused on the Automatic Detection of Objects in the Track.\n The repository with the code is accessible from: https://github.com/oValach/RailSafeNet_DT \n\nUpload an image with a scene including rail track and get a processed image with marked rail critical areas and detected and classified objects."
 )
 
+example_images = gr.Markdown(
+    """
+    ## Example input
+    Here are two example images that you can use:
+    """
+)
+
+example_image1 = gr.Image(value='rs00039.jpg', type='file', label="Example Image 1")
+example_image2 = gr.Image(value='rs00042.jpg', type='file', label="Example Image 2")
+
+# Combine the interface and example images
+app = gr.Blocks()
+
+with app:
+    iface.render()
+    example_images.render()
+    example_image1.render()
+    example_image2.render()
+
 # Launch the interface
 if __name__ == "__main__":
-    iface.launch()
+    app.launch()