apolinario commited on
Commit
16489cb
1 Parent(s): a17b7c9
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -13,13 +13,13 @@ 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)[0]
17
  print(image)
18
  return(image)
19
 
20
 
21
  def text2image_guided(text):
22
- image = guided(text,'',10,600,0,0,0,random.randint(0,2147483647),'',50,32)
23
  print(image)
24
  image = image[0]
25
  return(image)
@@ -49,6 +49,7 @@ with block:
49
  with gr.Column():
50
  with gr.Row():
51
  image = gr.outputs.Image()
 
52
  print(which_tab)
53
  get_image_latent.click(text2image_latent, inputs=[text,steps,width,height,images,diversity], outputs=image)
54
  get_image_rudalle.click(text2image_rudalle, inputs=[text,aspect,model], outputs=image)
 
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)
 
49
  with gr.Column():
50
  with gr.Row():
51
  image = gr.outputs.Image()
52
+
53
  print(which_tab)
54
  get_image_latent.click(text2image_latent, inputs=[text,steps,width,height,images,diversity], outputs=image)
55
  get_image_rudalle.click(text2image_rudalle, inputs=[text,aspect,model], outputs=image)