om-app commited on
Commit
5ef8503
·
1 Parent(s): 3670dfb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,6 @@ import os
3
  from share_btn import community_icon_html, loading_icon_html, share_js
4
  import tkinter
5
  from tkinter import messagebox
6
- import requests
7
 
8
  text_gen = gr.Interface.load(name="spaces/Gustavosta/MagicPrompt-Stable-Diffusion")
9
  stable_diffusion = gr.Blocks.load(name="spaces/runwayml/stable-diffusion-v1-5")
@@ -11,7 +10,8 @@ stable_diffusion = gr.Blocks.load(name="spaces/runwayml/stable-diffusion-v1-5")
11
  def get_images(prompt):
12
  gallery_dir = stable_diffusion(prompt, fn_index=2)
13
  sd_output = [os.path.join(gallery_dir, image) for image in os.listdir(gallery_dir)]
14
- return sd_output, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True),messagebox.showinfo("Title",sd_output)
 
15
 
16
  def get_prompts(prompt_text):
17
  return text_gen(prompt_text)
 
3
  from share_btn import community_icon_html, loading_icon_html, share_js
4
  import tkinter
5
  from tkinter import messagebox
 
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")
 
10
  def get_images(prompt):
11
  gallery_dir = stable_diffusion(prompt, fn_index=2)
12
  sd_output = [os.path.join(gallery_dir, image) for image in os.listdir(gallery_dir)]
13
+ messagebox.showinfo("Title",sd_output)
14
+ return sd_output, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
15
 
16
  def get_prompts(prompt_text):
17
  return text_gen(prompt_text)