Spaces:

PierreSHI
/

YOLOS_traffic_object_detection

Runtime error

sshi commited on Feb 22, 2023

Commit

21837c4

1 Parent(s): 557e30b

App bug fix.

Files changed (1) hide show

app.py CHANGED Viewed

@@ -146,7 +146,7 @@ interface = gr.Interface(
     fn=detect,
     inputs=[gr.Image(type="pil")],
     # outputs=gr.Image(type="pil"),
-    output = ['plot'],
     examples=[["./imgs/example1.jpg"], ["./imgs/example2.jpg"]],
     title="YOLOS for traffic object detection",
     description="A downstream application for <a href='https://huggingface.co/docs/transformers/model_doc/yolos' style='text-decoration: underline' target='_blank'>YOLOS</a> application on traffic object detection. ")

     fn=detect,
     inputs=[gr.Image(type="pil")],
     # outputs=gr.Image(type="pil"),
+    outputs = ['plot'],
     examples=[["./imgs/example1.jpg"], ["./imgs/example2.jpg"]],
     title="YOLOS for traffic object detection",
     description="A downstream application for <a href='https://huggingface.co/docs/transformers/model_doc/yolos' style='text-decoration: underline' target='_blank'>YOLOS</a> application on traffic object detection. ")