Update app.py

app.py

@@ -20,13 +20,17 @@ def demo_process(input_pdf):
     
     with open(os.path.join(temp_dir, image_name), "wb") as f:
         f.write(image_bytes)
-    
+
+    zip_path = "images.zip"
     image_path = os.path.join(temp_dir, image_name)
     
+    with zipfile.ZipFile(zip_path, "w") as zipf:
+        zipf.write(os.path.join(temp_dir, image_name), image_name)
+        
     os.remove(image_path)
     os.rmdir(temp_dir)
     
-    return image_path
+    return zip_path
 
 # task_name = "SGSInvoice"
 # task_prompt = f"<s_{task_name}>"