Update app.py

app.py

@@ -1,24 +1,29 @@
 import gradio as gr
 from ultralytics import YOLO
-import os
+import cv2
 
-# Load the model
-model = YOLO("best.pt")  # assumes best.pt is in the same folder
+# Load your custom-trained model
+model = YOLO("best.pt")  # Make sure best.pt is in the same directory
 
 # Inference function
-def predict(image):
+def detect_objects(image):
     results = model(image)
-    annotated_frame = results[0].plot()  # draw bounding boxes
-    return annotated_frame
+    annotated_image = results[0].plot()  # Draw bounding boxes
+    return annotated_image
+
+# Gradio Interface
+description = """
+📉 **Panel Fault D C Detection**  
+Upload an image to detect panel faults using a YOLOv8 model trained with [Roboflow](https://universe.roboflow.com/naveen-kumar-9emxl/my-project-1-scnhw).
+"""
 
-# Gradio UI
 demo = gr.Interface(
-    fn=predict,
+    fn=detect_objects,
     inputs=gr.Image(type="pil"),
     outputs=gr.Image(type="pil"),
-    title="YOLOv8 Object Detection",
-    description="Upload an image and see YOLOv8 detect objects using your custom-trained model."
+    title="Panel Fault D C",
+    description=description
 )
 
-# Launch app
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()