Update app.py

app.py

@@ -28,10 +28,17 @@ def extract_text(image, processor, model):
         inputs = processor(images=image_resized, return_tensors="pt", do_resize=False).to("cpu")
         pixel_values = inputs.get("pixel_values")
 
+        # Debugging: Check the pixel_values tensor shape
+        st.write(f"Pixel Values Shape: {pixel_values.shape}")
+
         # Check if pixel values are valid
         if pixel_values is None or pixel_values.shape[0] == 0:
             raise ValueError("Preprocessing failed: Empty tensor generated for image.")
 
+        # Additional check to ensure it has expected shape
+        if pixel_values.shape[0] != 81 or pixel_values.shape[1] != 2048:
+            raise ValueError(f"Unexpected tensor shape: {pixel_values.shape}. Expected shape: [batch_size, 2048].")
+
         progress_bar.progress(50)  # Step 2: Image preprocessed
         time.sleep(0.5)