Update app.py

app.py

@@ -70,7 +70,15 @@ def detect_document(image):
     detected_classes = set()  
     labels = []
     bounding_boxes = []
-
+    
+    if isinstance(image, np.ndarray):
+        if image.dtype != np.uint8:
+            image = (image * 255).clip(0, 255).astype(np.uint8)  # Convert float to uint8
+        
+        # Ensure correct shape (H, W, C)
+        if image.shape[0] == 1 and image.shape[1] == 1:
+            image = np.squeeze(image)
+            
     for result in results:
         for box in result.boxes:
             x1, y1, x2, y2 = map(int, box.xyxy[0])