commit

app.py

@@ -32,14 +32,15 @@ def image_fn(
         conf_threshold: Confidence threshold
         iou_threshold: IOU threshold
     Returns:
-        Rendered image
+        Rendered image as PIL image object
     """
 
     model = yolov5.load(model_path, device="cpu", hf_model=True, trace=False)
     model.conf = conf_threshold
     model.iou = iou_threshold
     results = model([image], size=image_size)
-    return results.render()[0]
+    return results.render()[0].astype('uint8')
+