Upload 3 files

app.py

@@ -0,0 +1,28 @@
+import cv2
+import numpy as np
+import gradio as gr
+from ultralytics import YOLO
+
+# Load trained model
+model = YOLO("E:\\job related docs\\thegame\\yolov8\\yolov8_model.pt")  # or "yolov8s.onnx"
+
+def detect_defects(image):
+    # Run inference
+    results = model(image)
+    # Plot results
+    annotated_img = results[0].plot(line_width=2)
+    return annotated_img
+
+# Gradio UI
+title = "🔍 Steel Surface Defect Detector (YOLOv8)"
+description = "Upload a steel surface image to detect defects (crazing, scratches, etc.)."
+examples = ["defect_example1.jpg", "defect_example2.jpg"]
+
+gr.Interface(
+    fn=detect_defects,
+    inputs=gr.Image(type="numpy"),
+    outputs=gr.Image(type="numpy"),
+    title=title,
+    description=description,
+    examples=examples,
+).launch()

requirements.txt

@@ -0,0 +1,5 @@
+ultralytics
+gradio
+numpy
+opencv-python
+onnxruntime

yolov8_model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ea9b9dd4ef4fd2786bbf13ccbe86d9014e545c788afc3eb5e6bd1f9f66e4e830
+size 6266171