Update app.py

app.py

@@ -13,10 +13,10 @@ def V_SPACE(lines):
     for _ in range(lines):
         st.write(' ')
 
-def image_download_button(pil_image+"/?target=external", filename: str, fmt: str, label="Download"):
+def image_download_button(pil_image, filename: str, fmt: str, label="Download"):
     if fmt not in ["jpg", "png"]:
         raise Exception(f"Unknown image format (Available: {fmt} - case sensitive)")
-
+   pil_image = pil_image+"/?target=external"
     pil_format = "JPEG" if fmt == "jpg" else "PNG"
     file_format = "jpg" if fmt == "jpg" else "png"
     mime = "image/jpeg/?target=external" if fmt == "jpg" else "image/png/?target=external"