apolinario commited on
Commit
4e01792
1 Parent(s): 16489cb

remove second button

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -13,13 +13,16 @@ def text2image_latent(text,steps,width,height,images,diversity):
13
  return(image)
14
 
15
  def text2image_rudalle(text,aspect,model):
16
- image = rudalle(text,aspect,model)
 
 
 
17
  print(image)
18
  return(image)
19
 
20
 
21
  def text2image_guided(text):
22
- image = guided(text,None,10,600,0,0,0,random.randint(0,2147483647),None,50,32)
23
  print(image)
24
  image = image[0]
25
  return(image)
@@ -45,7 +48,7 @@ with block:
45
  with gr.Tab("guided"):
46
  which_tab = "Guided Diffusion"
47
  get_image_guided = gr.Button("Generate Image")
48
- get_image = gr.Button("Generate Image")
49
  with gr.Column():
50
  with gr.Row():
51
  image = gr.outputs.Image()
 
13
  return(image)
14
 
15
  def text2image_rudalle(text,aspect,model):
16
+ print(text)
17
+ print(aspect)
18
+ print(model)
19
+ image = rudalle(text,aspect,model)[0]
20
  print(image)
21
  return(image)
22
 
23
 
24
  def text2image_guided(text):
25
+ image = guided(text, None, 10, 600, 0, 0, 0, random.randint(0,2147483647), None, 50, 32)
26
  print(image)
27
  image = image[0]
28
  return(image)
 
48
  with gr.Tab("guided"):
49
  which_tab = "Guided Diffusion"
50
  get_image_guided = gr.Button("Generate Image")
51
+ #get_image = gr.Button("Generate Image")
52
  with gr.Column():
53
  with gr.Row():
54
  image = gr.outputs.Image()