Spaces:
Running
Running
Commit
·
a0c06ea
1
Parent(s):
d163e81
Update app.py
Browse files
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)
|