mamechin
commited on
Commit
•
a02f75c
1
Parent(s):
70c3ac5
v1
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from ultralytics import YOLO
|
|
3 |
import torch
|
4 |
# model = YOLO('best.pt')
|
5 |
path = 'best.pt'
|
6 |
-
model = torch.hub.load("WongKinYiu/yolov7","custom",
|
7 |
|
8 |
def predict(input_image):
|
9 |
"""
|
|
|
3 |
import torch
|
4 |
# model = YOLO('best.pt')
|
5 |
path = 'best.pt'
|
6 |
+
model = torch.hub.load("WongKinYiu/yolov7","custom",path,trust_repo=True)
|
7 |
|
8 |
def predict(input_image):
|
9 |
"""
|