Update app.py

app.py

@@ -48,8 +48,22 @@ def normalize_2d(matrix):
 def detect(imagepath):
 
     pix=model.predict(imagepath, confidence=40, overlap=30)
-    #pix=np.array(pix)
-    return pix
+    pix=np.array(pix)
+    img=cv2.imread(imagepath)
+
+    for a,b,c,d,_,_,_,_ in pix.items():
+        cv2.rectangle(
+            img,
+            (a,b),
+            (a+c,b+d),
+            color=(0, 0, 255),
+            thickness=2,
+            lineType=cv2.LINE_AA)
+
+    return cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+
+    
+        
     
 def predict(img):
 
@@ -93,7 +107,7 @@ inputs_image = [
 ]
 
 outputs_image = [
-    gr.components.Image(type="pil", label="Output Image"),
+    gr.components.Image(type="numpy", label="Output Image"),
 ]
 
 # Create the Gradio demo