jiayueru commited on
Commit
78c154b
Β·
1 Parent(s): cd0ef8f
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -19,6 +19,8 @@ pipe.to('cuda')
19
  def generate(prompt):
20
  return pipe(prompt).images
21
 
 
 
22
  gr.Interface(
23
  fn=generate,
24
  inputs=gr.Text(),
 
19
  def generate(prompt):
20
  return pipe(prompt).images
21
 
22
+ gr.Image()
23
+
24
  gr.Interface(
25
  fn=generate,
26
  inputs=gr.Text(),