om-app commited on
Commit
5dc70a0
·
1 Parent(s): 8b934bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -8
app.py CHANGED
@@ -1,23 +1,18 @@
1
  import gradio as gr
2
  import os
3
  from share_btn import community_icon_html, loading_icon_html, share_js
4
- import urllib.request
5
- import shutil
6
 
7
  text_gen = gr.Interface.load(name="spaces/Gustavosta/MagicPrompt-Stable-Diffusion")
8
  stable_diffusion = gr.Blocks.load(name="spaces/runwayml/stable-diffusion-v1-5")
9
 
10
 
11
- def imp():
12
- url = "https://i.imgur.com/oer58GE.png"
13
- output_file = "save_this_name.png"
14
- with urllib.request.urlopen(url) as response, open(output_file, 'wb') as out_file:
15
- shutil.copyfileobj(response, out_file)
16
 
17
  def get_images(prompt):
18
  gallery_dir = stable_diffusion(prompt, fn_index=2)
19
  sd_output = [os.path.join(gallery_dir, image) for image in os.listdir(gallery_dir)]
20
- imp()
21
  return sd_output, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
22
 
23
  def get_prompts(prompt_text):
 
1
  import gradio as gr
2
  import os
3
  from share_btn import community_icon_html, loading_icon_html, share_js
4
+
 
5
 
6
  text_gen = gr.Interface.load(name="spaces/Gustavosta/MagicPrompt-Stable-Diffusion")
7
  stable_diffusion = gr.Blocks.load(name="spaces/runwayml/stable-diffusion-v1-5")
8
 
9
 
10
+
 
 
 
 
11
 
12
  def get_images(prompt):
13
  gallery_dir = stable_diffusion(prompt, fn_index=2)
14
  sd_output = [os.path.join(gallery_dir, image) for image in os.listdir(gallery_dir)]
15
+
16
  return sd_output, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
17
 
18
  def get_prompts(prompt_text):