Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -102,7 +102,7 @@ import gradio as gr
|
|
102 |
from ultralytics import YOLO
|
103 |
|
104 |
# Load the YOLO model (update the path to your fire detection model weights)
|
105 |
-
model = YOLO('
|
106 |
|
107 |
def detect_fire(frame):
|
108 |
# Convert the frame to RGB format (YOLO expects this format)
|
|
|
102 |
from ultralytics import YOLO
|
103 |
|
104 |
# Load the YOLO model (update the path to your fire detection model weights)
|
105 |
+
model = YOLO('best.pt') # Replace 'path/to/your/best.pt' with the actual path to your model file
|
106 |
|
107 |
def detect_fire(frame):
|
108 |
# Convert the frame to RGB format (YOLO expects this format)
|