App bug fix.

app.py

@@ -100,7 +100,7 @@ def rescale_bboxes(out_bbox, size):
     return b
 
 def plot_results(pil_img, prob, boxes):
-    fig = plt.figure(figsize=(16,10))
+    fig = plt.figure(figsize=(64,40))
     plt.imshow(pil_img)
     ax = plt.gca()
     colors = COLORS * 100
@@ -113,6 +113,7 @@ def plot_results(pil_img, prob, boxes):
         ax.text(xmin, ymin, text, fontsize=10,
                 bbox=dict(facecolor=c, alpha=0.5))
     plt.axis('off')
+    # return Image.frombytes('RGB', fig.canvas.get_width_height(),fig.canvas.tostring_rgb())
     return fig
 
 
@@ -144,7 +145,8 @@ def detect(img):
 interface = gr.Interface(
     fn=detect,
     inputs=[gr.Image(type="pil")], 
-    outputs=['plot'],
+    # 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. ")