run

app.py

@@ -39,7 +39,6 @@
 
 # if __name__ == "__main__":
 #     demo.launch()
-
 import gradio as gr
 import cv2
 import numpy as np
@@ -49,6 +48,8 @@ import gc
 from inspector_engine import AdvancedBlockInspector
 
 # Initialize engine with lazy loading
+# Note: HF Spaces will run this on startup. 
+# We use the local model file provided in the repository.
 inspector = AdvancedBlockInspector(yolo_model_path='yolo26n-obb.pt')
 
 def inspect(image):
@@ -93,26 +94,29 @@ def inspect(image):
         print(error_msg)
         return None, {"error": str(e)}
 
-# Create Gradio Interface
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# TMTL Industrial Inspector")
-    gr.Markdown("Remote AI Inference Engine for Saddle Defect Detection")
+# Create Gradio Interface with a premium theme
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="indigo")) as demo:
+    gr.Markdown("# 🔍 TMTL Industrial Inspector")
+    gr.Markdown("### Remote AI Inference Engine for Saddle Defect Detection")
     
     with gr.Row():
-        with gr.Column():
+        with gr.Column(scale=1):
             input_img = gr.Image(type="numpy", label="Source Image")
-            btn = gr.Button("Run Analysis", variant="primary")
+            btn = gr.Button("🚀 Run Analysis", variant="primary")
         
-        with gr.Column():
+        with gr.Column(scale=1):
             output_img = gr.Image(type="numpy", label="AI Visualization")
             output_json = gr.JSON(label="Detailed Analysis")
     
-    # Wire up the button with API name HERE
+    gr.Markdown("---")
+    gr.Markdown("© 2026 TMTL AI Solutions | Precision Inspection System")
+    
+    # Wire up the button with API name
     btn.click(
         fn=inspect, 
         inputs=input_img, 
         outputs=[output_img, output_json],
-        api_name="predict"  # MOVED HERE - this is the correct location
+        api_name="predict"
     )
 
 if __name__ == "__main__":
@@ -122,7 +126,5 @@ if __name__ == "__main__":
     ).launch(
         server_name="0.0.0.0",
         server_port=7860,
-        show_error=True,
-        share=False,
-        show_api=True
-    )
+        show_error=True
+    )