GenAIJake commited on
Commit
be12123
·
verified ·
1 Parent(s): 2a553a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -4
app.py CHANGED
@@ -7,9 +7,9 @@ def generate_image(prompt, negative_prompt="", seed=42, width=512, height=512, g
7
  pass
8
 
9
  examples = [
10
- ["d3xt3r dachshund as a camp counselor in the woods."],
11
- ["d3xt3r dachshund dressed as batman"],
12
- ["d3xt3r dachshund in a suit and tie"],
13
  ]
14
 
15
  demo = gr.load("models/GenAIJake/d3xt3r")
@@ -17,7 +17,22 @@ demo = gr.load("models/GenAIJake/d3xt3r")
17
  # Customize the interface
18
  with demo:
19
  gr.Markdown("# D3XT3R Dachshund Image Generator")
20
- gr.Examples(examples, inputs=demo.input_components)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
21
 
22
  # Add GPUZero functionality
23
  demo.queue()
 
7
  pass
8
 
9
  examples = [
10
+ ["d3xt3r dachshund as a camp counselor in the woods.", "", 42, 512, 512, 3.5, 20],
11
+ ["d3xt3r dachshund dressed as batman", "", 123, 512, 512, 7.0, 30],
12
+ ["d3xt3r dachshund in a suit and tie", "", 456, 768, 512, 5.0, 25],
13
  ]
14
 
15
  demo = gr.load("models/GenAIJake/d3xt3r")
 
17
  # Customize the interface
18
  with demo:
19
  gr.Markdown("# D3XT3R Dachshund Image Generator")
20
+ gr.Markdown("Generate images of D3XT3R the dachshund in various scenarios!")
21
+
22
+ with gr.Row():
23
+ with gr.Column():
24
+ prompt = gr.Textbox(label="Prompt", value="d3xt3r as a camp counselor in the woods.")
25
+ negative_prompt = gr.Textbox(label="Negative Prompt")
26
+ seed = gr.Slider(0, 1000000, label="Seed", value=42, step=1)
27
+ width = gr.Slider(256, 1024, label="Width", value=512, step=64)
28
+ height = gr.Slider(256, 1024, label="Height", value=512, step=64)
29
+ guidance_scale = gr.Slider(1, 20, label="Guidance Scale", value=3.5, step=0.1)
30
+ num_inference_steps = gr.Slider(1, 100, label="Number of Inference Steps", value=20, step=1)
31
+
32
+ with gr.Column():
33
+ output = gr.Image(label="Generated Image")
34
+
35
+ gr.Examples(examples, inputs=[prompt, negative_prompt, seed, width, height, guidance_scale, num_inference_steps], outputs=output)
36
 
37
  # Add GPUZero functionality
38
  demo.queue()