Update app.py

app.py

@@ -72,7 +72,7 @@ def main():
 
     if st.button(':blue[Generate Image]', use_container_width=True):
         with st.spinner('Generating Image...'):
-            result = generate_image(api_key, prompt, aspect_ratios, mode, models, seed, output_formats, strength, negative_prompt, uploaded_file, download)
+            result = generate_image(api_key, prompt, aspect_ratios, mode, models, seed, output_formats, strength, negative_prompt, uploaded_file)
             if result.status_code == 200:
                 image = Image.open(io.BytesIO(result.content))
                 st.image(image, use_column_width=True)
@@ -82,7 +82,7 @@ def main():
                 st.success(f'Image saved to {file_path}')
                 # Download generated image for Huggingface Spaces add-on
                 with open(file_path, "rb") as file:
-                    download = st.download_button(label=":green[Download Image]", data=file, file_name=file_path, mime=f"image/{output_formats}", use_container_width=True)
+                    st.download_button(label=":green[Download Image]", data=file, file_name=file_path, mime=f"image/{output_formats}", use_container_width=True)
             else:
                 st.error('Failed to generate image: ' + str(result.json()))