Update app.py

app.py

@@ -23,18 +23,19 @@ def predict(
         detection_threshold=detection_threshold
     )
     return samples[0]["img"], preds[0]
-
-def show_preds(input_image):
-    img = PIL.Image.fromarray(input_image, "RGB")
-    pred_dict = model_type.end2end_detect(img, model=model, 
-                                          class_map=class_map, 
-                                          detection_threshold=0.5,
-                                          display_label=False, 
-                                          display_bbox=True, 
-                                          return_img=True, 
-                                          font_size=16, 
-                                          label_color="#FF59D6")
-    return pred_dict["img"]
+    
+def show_preds(input_image, display_list, detection_threshold):
+    display_label = ("Label" in display_list)
+    display_bbox = ("BBox" in display_list)
+
+    if detection_threshold==0: detection_threshold=0.5
+
+    img, pred = predict(model=model, image=input_image, detection_threshold=detection_threshold)
+    # print(pred)
+    img = draw_pred(img=img, pred=pred, class_map=class_map, display_label=display_label, display_bbox=display_bbox)
+    img = PIL.Image.fromarray(img)
+    # print("Output Image: ", img.size, type(img))
+    return img
 
 # Populate examples in Gradio interface
 examples = [