chauhuynh90 commited on
Commit
a0c06ea
·
1 Parent(s): d163e81

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ from PIL import ImageFont
6
  import cv2
7
  import numpy as np
8
 
9
- model = torch.hub.load('ultralytics/yolov5', 'custom', path='yolov5n_rebar_kaggle.pt')
10
 
11
  def yolo(im, conf, iou, size=640):
12
 
@@ -57,7 +57,7 @@ outputs = [outputs_image, outputs_text]
57
  title = "OBJECT COUNTING"
58
  description = "Object counting demo. Upload an image or click an example image to use. You can select the area to count by drawing a closed area on the input image."
59
  article = "<p style='text-align: center'>Counting objects in image</a></p>"
60
- examples = [['S__275668998.jpg']]
61
 
62
  gr.Interface(yolo, inputs, outputs, title=title, description=description, article=article, examples=examples, analytics_enabled=False).launch(
63
  debug=True)#, share=True)
 
6
  import cv2
7
  import numpy as np
8
 
9
+ model = torch.hub.load('ultralytics/yolov5', 'custom', path='model/yolov5n_rebar_kaggle.pt')
10
 
11
  def yolo(im, conf, iou, size=640):
12
 
 
57
  title = "OBJECT COUNTING"
58
  description = "Object counting demo. Upload an image or click an example image to use. You can select the area to count by drawing a closed area on the input image."
59
  article = "<p style='text-align: center'>Counting objects in image</a></p>"
60
+ examples = [['images/S__275668998.jpg']]
61
 
62
  gr.Interface(yolo, inputs, outputs, title=title, description=description, article=article, examples=examples, analytics_enabled=False).launch(
63
  debug=True)#, share=True)