mamechin commited on
Commit
a02f75c
1 Parent(s): 70c3ac5
Files changed (1) hide show
  1. app.py +1 -1
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",f"{path}",trust_repo=True)
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
  """