Update app.py

app.py

@@ -5,6 +5,7 @@ import pyarrow.parquet as pq
 from PIL import Image
 import io
 import base64
+import tempfile
 
 def image_to_parquet(files):
     # List to store image data
@@ -28,15 +29,12 @@ def image_to_parquet(files):
     table = pa.Table.from_pandas(df)
     
     # Save table as Parquet file
-    parquet_buffer = io.BytesIO()
-    pq.write_table(table, parquet_buffer)
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".parquet") as tmp_file:
+        pq.write_table(table, tmp_file)
+        parquet_file_path = tmp_file.name
     
-    # Return Parquet file
-    parquet_buffer.seek(0)
-    return parquet_buffer
-
-def download_parquet(file):
-    return file
+    # Return Parquet file path
+    return parquet_file_path
 
 # Gradio interface
 with gr.Blocks() as demo:
@@ -48,4 +46,4 @@ with gr.Blocks() as demo:
     
     convert_button.click(fn=image_to_parquet, inputs=[image_input], outputs=[download_button])
 
-demo.launch(share=True)
+demo.launch()