Update app.py

app.py

@@ -15,6 +15,7 @@ import numpy as np
 
 import subprocess
 import tempfile
+import os
 
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
@@ -225,9 +226,10 @@ def process_image(image, task_prompt, text_input=None, model_id='microsoft/Flore
 # Function to convert the output text to a downloadable text file
 def save_text_as_file(output_text):
     # Use a temporary file to store the output text for download
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".txt", mode="w") as tmp_file:
-        tmp_file.write(output_text)
-        return tmp_file.name
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".txt")
+    with open(temp_file.name, "w") as f:
+        f.write(output_text)
+    return temp_file.name
 
 css = """
   #output {
@@ -285,6 +287,6 @@ with gr.Blocks(css=css) as demo:
         submit_btn.click(process_image, [input_img, task_prompt, text_input, model_selector], [output_text, output_img])
         
         # Add functionality to generate and download text file
-        download_btn.upload(save_text_as_file, inputs=output_text, outputs=download_btn)
+        download_btn.update(save_text_as_file, inputs=output_text, outputs=download_btn)
 
-demo.launch(debug=True)
+demo.launch(debug=True)