avanish07 commited on
Commit
20087e5
1 Parent(s): 618ca97

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import os
5
 
6
  from ultralytics import YOLO
7
 
8
- model = YOLO('best.pt')
9
  image_paths = ['pothole_example.jpg', 'pothole_screenshot.png']
10
 
11
  def show_preds_image(image):
 
5
 
6
  from ultralytics import YOLO
7
 
8
+ model = YOLO('best_gd.pt')
9
  image_paths = ['pothole_example.jpg', 'pothole_screenshot.png']
10
 
11
  def show_preds_image(image):