ArchiMathur commited on
Commit
e2331ce
·
verified ·
1 Parent(s): 8f43242

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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('path/to/your/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)
 
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)