Update app.py

app.py

@@ -27,7 +27,7 @@ CLASS_COLORS = {
 # Global model variable
 model = None
 
-def load_model(model_path: str = "model/full_29.pth"):
+def load_model(model_path: str = "model.pth"):
     """Load RF-DETR model"""
     global model
     if model is None:
@@ -156,24 +156,23 @@ def detect_ui_elements(
     return annotated_pil, summary_text
 
 # Gradio interface
-with gr.Blocks(title="RF-DETR UI Element Detector", theme=gr.themes.Soft()) as demo:
+with gr.Blocks(title="CU-1 UI Element Detector", theme=gr.themes.Soft()) as demo:
 
     gr.Markdown("""
-    # 🎯 RF-DETR UI Element Detector
+    # CU-1 UI Element Detector
 
-    Upload a screenshot or UI mockup to automatically detect interactive elements.
-    This model identifies 8 types of UI components: buttons, fields, headings, iframes, images, labels, links, and text.
+    Upload a screenshot or UI mockup to automatically detect elements.
     """)
 
     with gr.Row():
         with gr.Column(scale=1):
             input_image = gr.Image(
                 type="pil",
-                label="📤 Upload Screenshot",
+                label="Upload Screenshot",
                 height=400
             )
 
-            with gr.Accordion("⚙️ Detection Settings", open=True):
+            with gr.Accordion("Detection Settings", open=True):
                 confidence_slider = gr.Slider(
                     minimum=0.1,
                     maximum=0.9,
@@ -191,38 +190,17 @@ with gr.Blocks(title="RF-DETR UI Element Detector", theme=gr.themes.Soft()) as d
                     label="Box Line Thickness"
                 )
 
-            detect_button = gr.Button("🔍 Detect Elements", variant="primary", size="lg")
-
-            gr.Markdown("""
-            ### 📊 Detected Classes:
-            - 🟢 **button** - Interactive buttons
-            - 🔵 **field** - Input fields
-            - 🟣 **heading** - Headers and titles
-            - 🟡 **iframe** - Embedded frames
-            - 🟠 **image** - Images and icons
-            - 🔷 **label** - Text labels
-            - 🔴 **link** - Hyperlinks
-            - ⚪ **text** - Plain text
-            """)
+            detect_button = gr.Button("Detect Elements", variant="primary", size="lg")
 
         with gr.Column(scale=1):
             output_image = gr.Image(
                 type="pil",
-                label="🎨 Detected Elements",
+                label="Detected Elements",
                 height=400
             )
 
-            summary_output = gr.Markdown(label="📋 Detection Summary")
+            summary_output = gr.Markdown(label="Detection Summary")
 
-    # Examples
-    gr.Markdown("### 💡 Try with example images:")
-    gr.Examples(
-        examples=[
-            # Add example image paths here if available
-        ],
-        inputs=input_image,
-        label="Example Screenshots"
-    )
 
     # Connect button
     detect_button.click(
@@ -231,11 +209,6 @@ with gr.Blocks(title="RF-DETR UI Element Detector", theme=gr.themes.Soft()) as d
         outputs=[output_image, summary_output]
     )
 
-    gr.Markdown("""
-    ---
-    **Model:** RF-DETR Medium (Resolution: 1600px) | **Framework:** PyTorch
-    """)
-
 # Launch
 if __name__ == "__main__":
     demo.queue().launch(share=False)