test

app.py

@@ -205,7 +205,7 @@ def predict_pipeline(img_input,
     md_results = predict_md(img_input, 
                             mega_model_input,
                             size=640) #Image.fromarray(results.imgs[0])
-    pdb.set_trace()
+    #pdb.set_trace()
     ################################################################
     # Obtain animal crops for bboxes with confidence above th
 
@@ -243,7 +243,7 @@ def predict_pipeline(img_input,
                                          dlc_proc)
 
         # Produce final image
-        img_background = Image.fromarray(md_results.imgs[0]) # img_input or Image.fromarray(md_results.imgs[0])? 
+        img_background = Image.fromarray(md_results.ims[0]) # img_input or Image.fromarray(md_results.imgs[0])? 
         # Image.fromarray(md_results.imgs[0]) --> (640, 479)
         # img_input.size ---> (259, 194)
         # pdb.set_trace()