Update app.py

app.py

@@ -159,7 +159,7 @@ HTML_TEMPLATE = """
 
 # The rest of the Python code remains the same
 def yolov9_inference(img_path, image_size, conf_threshold, iou_threshold):
-    model = yolov9.load('./best.pt')
+    model = yolov9.load('./best.pt') # Load your trained model
     model.conf = conf_threshold
     model.iou = iou_threshold
     results = model(img_path, size=image_size)
@@ -167,7 +167,7 @@ def yolov9_inference(img_path, image_size, conf_threshold, iou_threshold):
     return output[0]
 
 def app():
-    with gr.Blocks() as demo:
+    with gr.Blocks(theme=gr.themes.Soft()) as demo: # Added a theme here
         gr.HTML(HTML_TEMPLATE)
         
         with gr.Row():
@@ -189,8 +189,8 @@ def app():
         
         gr.Examples(
             examples=[
-                ["./openmanhole.jpg", 640, 0.4, 0.5],
-                ["./images.jpeg", 640, 0.4, 0.5],
+                ["./openmanhole.jpg", 640, 0.4, 0.5], # Add your example images
+                ["./images.jpeg", 640, 0.4, 0.5],  # Add your example images
             ],
             fn=yolov9_inference,
             inputs=[img_path, image_size, conf_threshold, iou_threshold],
@@ -200,16 +200,9 @@ def app():
     
     return demo
 
-css = """
-/* You can add any additional CSS here to style Gradio components */
-.gradio-container {
-    max-width: 900px !important;
-    margin-left: auto !important;
-    margin-right: auto !important;
-}
-"""
+# Removed the separate CSS variable and added theme to gr.Blocks
 
-demo = gr.Blocks(css=css)
+demo = gr.Blocks() # Moved the theme application here
 
 with demo:
     app()