Update app.py

app.py

@@ -8,6 +8,15 @@ from ultralytics import YOLO
 # Download model if not present
 model_path ="best.pt"
 
+# Load YOLOv8 model
+model = YOLO("best.pt")
+
+def detect(image):
+    results = model(image)
+    annotated_image = results[0].plot()  # draw boxes
+    return annotated_image
+
+
 def detect_faults(video_path):
     try:
         # Generate unique run directory
@@ -50,18 +59,7 @@ def detect_faults(video_path):
         cap.release()
         out.release()
 
-        # Return the output video path
-        if os.path.exists(output_path):
-            print(f"✅ Output saved at: {output_path}")
-            return output_path
-        else:
-            print("❌ Output video not found.")
-            return "Error: Annotated video not found."
-
-    except Exception as e:
-        print(f"❌ Exception during fault detection: {e}")
-        return "Error during video fault detection."
-
+        
 def detect_faults_in_image(image):
     try:
         results = model(image)
@@ -80,15 +78,19 @@ video_ui = gr.Interface(
     description="Upload a drone video of solar panels. The model detects faults and returns an annotated video."
 )
 
-image_ui = gr.Interface(
-    fn=detect_faults_in_image,
-    inputs=gr.Image(type="numpy", label="Upload Solar Panel Image"),
-    outputs=gr.Image(label="Detected Faults in Image"),
-    title="Solar Panel Fault Detection (Image)",
-    description="Upload an image of a solar panel. The model highlights any detected faults."
+# Gradio Interface
+demo = gr.Interface(
+    fn=detect,
+    inputs=gr.Image(type="filepath", label="Upload an Image"),
+    outputs=gr.Image(type="numpy", label="Detected Output"),
+    title="Solar Panel Fault Detector - YOLOv8",
+    description="Upload a drone image of solar panels. YOLOv8 will detect any faults."
 )
 
-app = gr.TabbedInterface([video_ui, image_ui], ["Video Detection", "Image Detection"])
+demo.launch()
+
+
+app = gr.TabbedInterface([image_ui, video_ui], ["Image Detection"], "Video Detection")
 
 if __name__ == "__main__":
     app.launch()