paugar commited on
Commit
0d3a1e8
·
verified ·
1 Parent(s): bd78031

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -26,8 +26,7 @@ def load_img (filename):
26
  def process_img(image):
27
 
28
  with torch.no_grad():
29
- result = model.predict(source=image)
30
-
31
  if len(result[0].boxes)>0:
32
  vis = plots.plot_detection(image,boxes=result[0].boxes.boxes,
33
  class_names=class_names, min_score=0.2)
 
26
  def process_img(image):
27
 
28
  with torch.no_grad():
29
+ result = model(source=image)
 
30
  if len(result[0].boxes)>0:
31
  vis = plots.plot_detection(image,boxes=result[0].boxes.boxes,
32
  class_names=class_names, min_score=0.2)