om-app commited on
Commit
2b791e4
·
1 Parent(s): ffddc9e

Update app.py

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