Update app.py

app.py

@@ -4,6 +4,12 @@ from PIL import Image
 import os 
 import yolov9
 
+import gradio as gr
+import torch
+from PIL import Image
+import os 
+import yolov9
+
 def yolov9_inference(img_path, image_size, conf_threshold, iou_threshold):
     model = yolov9.load('./best.pt')
     model.conf = conf_threshold
@@ -25,11 +31,10 @@ def app():
         with gr.Row():
             with gr.Column(scale=1, min_width=300):
                 img_path = gr.Image(type="filepath", label="Upload Image")
-                with gr.Box():
-                    gr.Markdown("### Model Parameters")
-                    image_size = gr.Slider(label="Image Size", minimum=320, maximum=1280, step=32, value=640)
-                    conf_threshold = gr.Slider(label="Confidence Threshold", minimum=0.1, maximum=1.0, step=0.1, value=0.4)
-                    iou_threshold = gr.Slider(label="IoU Threshold", minimum=0.1, maximum=1.0, step=0.1, value=0.5)
+                gr.Markdown("### Model Parameters")
+                image_size = gr.Slider(label="Image Size", minimum=320, maximum=1280, step=32, value=640)
+                conf_threshold = gr.Slider(label="Confidence Threshold", minimum=0.1, maximum=1.0, step=0.1, value=0.4)
+                iou_threshold = gr.Slider(label="IoU Threshold", minimum=0.1, maximum=1.0, step=0.1, value=0.5)
                 yolov9_infer = gr.Button("Detect Manholes", variant="primary")
             
             with gr.Column(scale=1, min_width=300):
@@ -75,13 +80,6 @@ body {
 .gr-button:hover {
     background-color: #2980b9 !important;
 }
-.gr-box {
-    border-radius: 8px;
-    border: 1px solid #e0e0e0;
-    padding: 15px;
-    margin-top: 10px;
-    background-color: #f9f9f9;
-}
 .gr-form {
     background-color: white;
     padding: 15px;