Update app.py

app.py

@@ -39,7 +39,34 @@ url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
 
 
+#####################################################
+def predict_pipeline(img_input,
+                     mega_model_input,
+                     dlc_model_input_str,
+                     flag_dlc_only,
+                     flag_show_str_labels,
+                     bbox_likelihood_th,
+                     kpts_likelihood_th,
+                     font_style,
+                     font_size,
+                     keypt_color,
+                     marker_size,
+                     ):
 
+    if not flag_dlc_only:
+        ############################################################                                               
+        # ### Run Megadetector
+        md_results = predict_md(img_input, 
+                                MD_models_dict[mega_model_input], #mega_model_input,
+                                size=640) #Image.fromarray(results.imgs[0])
+
+        ################################################################
+        # Obtain animal crops for bboxes with confidence above th
+        list_crops = crop_animal_detections(img_input,
+                                            md_results,
+                                            bbox_likelihood_th)
+
+    ############################################################