brian25 commited on
Commit
b1fbed0
·
verified ·
1 Parent(s): 6e1d44e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ model = YOLO("best.pt")
6
 
7
  def greet(source):
8
  model = YOLO("best.pt")
9
- results = model.predict('/content/download.png', imgsz=1080, conf=0.25)
10
  result = results[0]
11
  return result.plot()
12
 
 
6
 
7
  def greet(source):
8
  model = YOLO("best.pt")
9
+ results = model.predict(source, imgsz=1080, conf=0.25)
10
  result = results[0]
11
  return result.plot()
12