siddharth060104 commited on
Commit
65430ae
·
verified ·
1 Parent(s): 478dfb4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -193,7 +193,7 @@ def main():
193
  height, width, _ = processed_image.shape
194
 
195
  # Initialize the YOLO model
196
- model = YOLO("standard.pt") # Replace with your trained model weights file
197
 
198
  # Run detection
199
  results = model.predict(source=processed_image, save=False, save_txt=False, conf=0.6)
 
193
  height, width, _ = processed_image.shape
194
 
195
  # Initialize the YOLO model
196
+ model = YOLO("fine_tuned_on_all_data.pt") # Replace with your trained model weights file
197
 
198
  # Run detection
199
  results = model.predict(source=processed_image, save=False, save_txt=False, conf=0.6)