Update app.py

app.py

@@ -134,6 +134,7 @@ def load_model():
             return False
     return True
 
+@spaces.GPU
 def perform_ocr(image):
     """Main OCR function with proper error handling."""
     try:
@@ -150,7 +151,17 @@ def perform_ocr(image):
         logger.error(f"Error in perform_ocr: {e}")
         return f"An error occurred: {str(e)}"
 
-@spaces.GPU
+def process_with_status(image):
+    """Process image and return result with status - moved outside to avoid pickling issues."""
+    if image is None:
+        return "Please upload an image.", "No image provided"
+    
+    try:
+        result = perform_ocr(image)
+        return result, "Processing completed successfully"
+    except Exception as e:
+        return f"Error: {str(e)}", f"Error occurred: {str(e)}"
+
 def create_interface():
     """Create the Gradio interface with proper configuration."""
     
@@ -236,16 +247,6 @@ def create_interface():
                     """)
         
         # Set up processing flow
-        def process_with_status(image):
-            if image is None:
-                return "Please upload an image.", "No image provided"
-            
-            try:
-                result = perform_ocr(image)
-                return result, "Processing completed successfully"
-            except Exception as e:
-                return f"Error: {str(e)}", f"Error occurred: {str(e)}"
-        
         submit_btn.click(
             fn=process_with_status, 
             inputs=image_input, 
@@ -273,4 +274,4 @@ if __name__ == "__main__":
         server_port=7860,
         share=False,
         debug=True
-    )
+    )