fixed input of create zip

app.py

@@ -269,15 +269,17 @@ with gr.Blocks(css=css) as demo:
         download_output = gr.File(label="Download")
 
     # Modify create_zip to use the state data
-    def create_zip(image, text_data, fname, ocr_result):
+    def create_zip(image, fname, ocr_result):
         # Validate inputs
-        if not fname or not image or not isinstance(image, (Image.Image, np.ndarray)):
+        if not fname or image is None:  # Changed the validation check
             return None
         
         try:
             # Convert numpy array to PIL Image if needed
             if isinstance(image, np.ndarray):
                 image = Image.fromarray(image)
+            elif not isinstance(image, Image.Image):
+                return None
             
             with tempfile.TemporaryDirectory() as temp_dir:
                 # Save image
@@ -324,7 +326,7 @@ with gr.Blocks(css=css) as demo:
     # Update the download button click handler to include state
     download_btn.click(
         create_zip,
-        inputs=[input_img, output_text, filename, ocr_state],
+        inputs=[input_img, filename, ocr_state],
         outputs=[download_output]
     )