Update app.py

app.py

@@ -35,26 +35,17 @@ def yolov9_inference(img_path):
 
     return output_image
 
-
 def app():
-    with gr.Blocks():
-        with gr.Row():
-            with gr.Column():
-                img_path = gr.Image(type="file", label="Upload Image")
-                inference_button = gr.Button(label="Run Inference")
-            with gr.Column():
-                output_image = gr.Image(label="Output Image")
-
-            inference_button.click(
-                fn=yolov9_inference,
-                inputs=[img_path],
-                outputs=[output_image],
-            )
-
-
-gradio_app = gr.Interface(
-    app,
-    title="YOLOv9 Inference",
-    description="Perform object detection using the YOLOv9 model.",
-)
+    return gr.Interface(
+        yolov9_inference,
+        inputs=gr.inputs.Image(type="pil", label="Upload Image"),
+        outputs="image",
+        title="YOLOv9 Inference",
+        description="Perform object detection using the YOLOv9 model.",
+        theme="compact"
+    )
+
+
+gradio_app = app()
 gradio_app.launch(debug=True)
+