App bug fix.

app.py

@@ -113,12 +113,11 @@ 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
 
 
 def generate_preds(processor, model, image):
     inputs = processor(images=image, return_tensors="pt").to(device)
-    # pixel_values = inputs.pixel_values.unsqueeze(0)
     preds = model(pixel_values=inputs.pixel_values)
     return preds
 
@@ -145,7 +144,7 @@ def detect(img):
 interface = gr.Interface(
     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. ")