matt-le-kat commited on
Commit
f61f64b
1 Parent(s): a832303

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -2
app.py CHANGED
@@ -18,8 +18,17 @@ def infer(prompt):
18
  images_list = pipe(
19
  [prompt])['sample']
20
  return images_list
21
- prompt = 'a perfect photo of a sunset on a Greek beach'
22
 
23
- intf = gr.Interface(fn = infer, inputs = prompt, outputs = gr.outputs.Image())
 
 
 
 
 
 
 
 
 
 
24
 
25
  intf.launch()
 
18
  images_list = pipe(
19
  [prompt])['sample']
20
  return images_list
 
21
 
22
+ text = gr.Textbox(
23
+ label="Enter your prompt",
24
+ show_label=False,
25
+ max_lines=1,
26
+ placeholder="Enter your prompt")
27
+
28
+ gallery = gr.Gallery(
29
+ label="Generated images",
30
+ show_label=False)
31
+
32
+ intf = gr.Interface(fn = infer, inputs = text, outputs = gallery)
33
 
34
  intf.launch()