Updated app

app.py

@@ -18,8 +18,10 @@ model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50")
 COLORS = [[0.000, 0.447, 0.741], [0.850, 0.325, 0.098], [0.929, 0.694, 0.125],
           [0.494, 0.184, 0.556], [0.466, 0.674, 0.188], [0.301, 0.745, 0.933]]
 
-def generate_output_figure(in_pil_img, in_results):
+def get_figure(in_pil_img, in_results):
     plt.figure(figsize=(16, 10))
+    plt.imshow(in_pil_img)
+    #pyplot.gcf()
     ax = plt.gca()
 
     for prediction in in_results:
@@ -44,7 +46,9 @@ def detect(image):
     with torch.no_grad():
         outputs = model(**encoding)
 
-    output_figure = generate_output_figure(image, outputs)
+    print(outputs)
+
+    output_figure = get_figure(image, outputs)
 
     buf = io.BytesIO()
     output_figure.savefig(buf, bbox_inches='tight')