Spaces:
Build error
Build error
removed fps and cuda indicator
Browse files
app.py
CHANGED
@@ -175,9 +175,7 @@ def process_img(model, image, labels, caption: bool = True):
|
|
175 |
cap = '{}'.format(label_name) if caption else ''
|
176 |
draw_caption(image_orig, (x1, y1, x2, y2), cap)
|
177 |
cv2.rectangle(image_orig, (x1, y1), (x2, y2), color=colors[label_name], thickness=2)
|
178 |
-
|
179 |
-
f"{'{:.1f}'.format(1 / float(elapsed_time))}{' cuda:' + str(torch.cuda.is_available()).lower()}",
|
180 |
-
fontScale=1, fontFace=cv2.FONT_HERSHEY_PLAIN, org=(10, 20), color=(0, 255, 0))
|
181 |
return image_orig
|
182 |
|
183 |
|
|
|
175 |
cap = '{}'.format(label_name) if caption else ''
|
176 |
draw_caption(image_orig, (x1, y1, x2, y2), cap)
|
177 |
cv2.rectangle(image_orig, (x1, y1), (x2, y2), color=colors[label_name], thickness=2)
|
178 |
+
|
|
|
|
|
179 |
return image_orig
|
180 |
|
181 |
|