Spaces:

PierreSHI
/

YOLOS_traffic_object_detection

Runtime error

sshi commited on Feb 22, 2023

Commit

ff116e7

1 Parent(s): 88c51e4

App bug fix.

Files changed (1) hide show

app.py CHANGED Viewed

@@ -115,7 +115,7 @@ def plot_results(pil_img, prob, boxes):
         #     fill=c)
         # ax.text(xmin, ymin, text, fontsize=10,
         #         bbox=dict(facecolor=c, alpha=0.5))
-    return return Image.fromarray(img[:,:,::-1])
     # return fig

         #     fill=c)
         # ax.text(xmin, ymin, text, fontsize=10,
         #         bbox=dict(facecolor=c, alpha=0.5))
+    return Image.fromarray(img[:,:,::-1])
     # return fig