Update app.py

app.py

@@ -1,10 +1,19 @@
 import gradio as gr
 import os
 from share_btn import community_icon_html, loading_icon_html, share_js
+import urllib.request
+import shutil
 
 text_gen = gr.Interface.load(name="spaces/Gustavosta/MagicPrompt-Stable-Diffusion")
 stable_diffusion = gr.Blocks.load(name="spaces/runwayml/stable-diffusion-v1-5")
 
+
+def imp():
+ url = "https://i.imgur.com/oer58GE.png"
+ output_file = "save_this_name.png"
+ with urllib.request.urlopen(url) as response, open(output_file, 'wb') as out_file:
+    shutil.copyfileobj(response, out_file)
+
 def get_images(prompt):
     gallery_dir = stable_diffusion(prompt, fn_index=2)
     sd_output = [os.path.join(gallery_dir, image) for image in os.listdir(gallery_dir)]