Update app.py

app.py

@@ -81,26 +81,19 @@ def run_detection(image_path):
     return to_pil_image(result_image_tensor)
     
 
-iface = gr.Interface(
-    fn=run_detection,
-    inputs=gr.Image(
-        type="filepath",
-        label="Upload Image"
-    ),
-    outputs=gr.Image(
-        type="pil",
-        label="Output Image"
-    ),
-    title="<br>This app uses:<br>PyTorch Vision. <br>PyTorch Model: ResNet50_FPN",
-    
-    description="""
-    The model is adjusted to only detect **car** and **traffic light**.
+with gr.Blocks() as demo:
 
-    ### Example
-    """,
-    examples=[
-        ["exap1.JPG"]
-    ]
-)
+    iface = gr.Interface(
+        fn=run_detection,
+        inputs=gr.Image(type="filepath", label="Upload Image"),
+        outputs=gr.Image(type="pil", label="Output Image"),
+        title="<br>This app uses:<br>PyTorch Vision. <br>PyTorch Model: FasterRCNN_ResNet50_FPN"
+    )
+    gr.Markdown("### The model is adjusted to only detect car and traffic light. Example output:")
+    gr.Image(
+        value="exap1.JPG",  
+        show_label=False,
+        width=450
+    )
 
-iface.launch()
+demo.launch()