Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ from ultralytics import YOLO
|
|
| 8 |
file_urls = [
|
| 9 |
#'https://www.dropbox.com/s/b5g97xo901zb3ds/pothole_example.jpg?dl=1',
|
| 10 |
#'https://www.dropbox.com/s/86uxlxxlm1iaexa/pothole_screenshot.png?dl=1',
|
| 11 |
-
'https://www.dropbox.com/s/7sjfwncffg8xej2/video_7.mp4?dl=1'
|
| 12 |
]
|
| 13 |
|
| 14 |
def download_file(url, save_name):
|
|
@@ -31,7 +31,7 @@ for i, url in enumerate(file_urls):
|
|
| 31 |
|
| 32 |
model = YOLO('yolov8_pcb.pt')
|
| 33 |
path = [['spur.jpg'], ['mouse.jpg']]
|
| 34 |
-
video_path = [['video.mp4']]
|
| 35 |
|
| 36 |
def show_preds_image(image_path):
|
| 37 |
image = cv2.imread(image_path)
|
|
|
|
| 8 |
file_urls = [
|
| 9 |
#'https://www.dropbox.com/s/b5g97xo901zb3ds/pothole_example.jpg?dl=1',
|
| 10 |
#'https://www.dropbox.com/s/86uxlxxlm1iaexa/pothole_screenshot.png?dl=1',
|
| 11 |
+
#'https://www.dropbox.com/s/7sjfwncffg8xej2/video_7.mp4?dl=1'
|
| 12 |
]
|
| 13 |
|
| 14 |
def download_file(url, save_name):
|
|
|
|
| 31 |
|
| 32 |
model = YOLO('yolov8_pcb.pt')
|
| 33 |
path = [['spur.jpg'], ['mouse.jpg']]
|
| 34 |
+
#video_path = [['video.mp4']]
|
| 35 |
|
| 36 |
def show_preds_image(image_path):
|
| 37 |
image = cv2.imread(image_path)
|