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)
|