Update app.py

app.py

@@ -51,7 +51,13 @@ def app():
         )
     
     return demo
-css = """
+import gradio as gr
+import torch
+from PIL import Image
+import os 
+import yolov9
+
+HTML_TEMPLATE = """
 <style>
     body {
         background: linear-gradient(135deg, #2c3e50, #34495e, #95a5a6);
@@ -185,6 +191,26 @@ css = """
         </div>
     </div>
 </div>
+"""
+
+# Your existing yolov9_inference function here
+def yolov9_inference(img_path, image_size, conf_threshold, iou_threshold):
+    # ... (your existing code)
+
+def app():
+    with gr.Blocks() as demo:
+        gr.HTML(HTML_TEMPLATE)
+        
+        # ... (rest of your existing app code)
+
+    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;
 }
 """