Update app.py

app.py

@@ -200,9 +200,11 @@ def predict(prompt, guidance, steps, seed=1231231):
             image_path = os.path.join("", filename)  # Specify your directory
             results.images[0].save(image_path)
             print(f"#Image saved as {image_path}")
-            encoded_image = encode_image(image)
-            html_link = f'<a href="data:image/png;base64,{encoded_image}" download="{filename}">Download Image</a>'
-            gr.HTML(html_link)
+            gr.File(image_path)
+            gr.Button(link=image_path)
+            # encoded_image = encode_image(image)
+            # html_link = f'<a href="data:image/png;base64,{encoded_image}" download="{filename}">Download Image</a>'
+            # gr.HTML(html_link)
     except:
         return results.images[0]