Files

app.py

@@ -0,0 +1,33 @@
+import gradio as gr
+
+from PIL import Image
+
+
+# Load the YOLOv8 model
+from ultralytics import YOLO
+model = YOLO("C:/Users/rater/Downloads/best.pt")
+
+# Define a function for inference
+def detect_objects(input_image):
+    # Convert Gradio input to a PIL image
+   
+    # Perform object detection
+    results = model(input_image)
+    for r in results:
+        im_array = r.plot()  # plot a BGR numpy array of predictions
+        im = Image.fromarray(im_array[..., ::-1])
+        return  im
+# Define the Gradio interface
+outputs = gr.outputs.Image(type="pil", label="Output Image")
+
+
+demo = gr.Interface(
+    detect_objects,
+    gr.Image(type="pil"),
+    outputs,
+   title="Burn Detection"
+  
+)
+
+demo.launch()
+

best.pt

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