Spaces:
Runtime error
Runtime error
sshi
commited on
Commit
•
e209888
1
Parent(s):
b3dd9ba
App bug fix.
Browse files
app.py
CHANGED
@@ -100,7 +100,7 @@ def rescale_bboxes(out_bbox, size):
|
|
100 |
return b
|
101 |
|
102 |
def plot_results(pil_img, prob, boxes):
|
103 |
-
fig = plt.figure(figsize=(
|
104 |
plt.imshow(pil_img)
|
105 |
ax = plt.gca()
|
106 |
colors = COLORS * 100
|
@@ -149,6 +149,6 @@ interface = gr.Interface(
|
|
149 |
outputs = ['plot'],
|
150 |
examples=[["./imgs/example1.jpg"], ["./imgs/example2.jpg"]],
|
151 |
title="YOLOS for traffic object detection",
|
152 |
-
description="A downstream application for <a href='https://huggingface.co/docs/transformers/model_doc/yolos' style='text-decoration: underline' target='_blank'>YOLOS</a>
|
153 |
|
154 |
interface.launch()
|
|
|
100 |
return b
|
101 |
|
102 |
def plot_results(pil_img, prob, boxes):
|
103 |
+
fig = plt.figure(figsize=(16,10), dpi=120)
|
104 |
plt.imshow(pil_img)
|
105 |
ax = plt.gca()
|
106 |
colors = COLORS * 100
|
|
|
149 |
outputs = ['plot'],
|
150 |
examples=[["./imgs/example1.jpg"], ["./imgs/example2.jpg"]],
|
151 |
title="YOLOS for traffic object detection",
|
152 |
+
description="A downstream application for <a href='https://huggingface.co/docs/transformers/model_doc/yolos' style='text-decoration: underline' target='_blank'>YOLOS</a> which can performe traffic object detection. ")
|
153 |
|
154 |
interface.launch()
|