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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -19
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.", "", 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,22 +17,7 @@ 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()
 
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
  # 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()