seawolf2357 commited on
Commit
d588de2
·
verified ·
1 Parent(s): d22e6d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -27
app.py CHANGED
@@ -30,8 +30,6 @@ def generate_image(prompt, seed=24, width=DEFAULT_WIDTH, height=DEFAULT_HEIGHT,
30
  seed = random.randint(0, MAX_SEED)
31
  generator = torch.Generator().manual_seed(int(float(seed)))
32
 
33
- start_time = time.time()
34
-
35
  img = pipe.generate_images(
36
  prompt=prompt,
37
  width=width,
@@ -39,8 +37,7 @@ def generate_image(prompt, seed=24, width=DEFAULT_WIDTH, height=DEFAULT_HEIGHT,
39
  num_inference_steps=DEFAULT_INFERENCE_STEPS,
40
  generator=generator
41
  )
42
- latency = f"Generation time: {(time.time()-start_time):.2f} seconds"
43
- return img, seed, latency
44
 
45
  # Example prompts
46
  examples = [
@@ -53,7 +50,6 @@ examples = [
53
  "A stylish young woman with long, wavy ombre hair and winged eyeliner, posing in front of a neon-lit city skyline at night. She's wearing a sleek black leather jacket over a sparkly crop top and holding up a holographic business card that says 'Invite' in futuristic font. The card reflects the colorful neon lights, creating a cyberpunk aesthetic.",
54
  ]
55
 
56
-
57
  css = """
58
  footer {visibility: hidden;}
59
  .container {max-width: 1200px; margin: auto; padding: 20px;}
@@ -62,7 +58,7 @@ footer {visibility: hidden;}
62
  border-radius: 10px;
63
  padding: 20px;
64
  margin-bottom: 20px;
65
- height: 400px; /* 고정된 높이 설정 */
66
  display: flex;
67
  flex-direction: column;
68
  }
@@ -93,19 +89,15 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
93
  placeholder="sexy woman & man , under wear, full body, sunday",
94
  lines=3,
95
  )
96
- with gr.Row():
97
- generateBtn = gr.Button("Generate Image", variant="primary")
98
- enhanceBtn = gr.Button("Enhance Image", variant="secondary")
99
-
100
- latency = gr.Text(label="Generation Time")
101
 
102
- with gr.Accordion("Advanced Options", open=False, elem_classes="advanced-options"):
103
- with gr.Row():
104
- seed = gr.Number(label="Seed", value=42)
105
- randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
106
- with gr.Row():
107
- width = gr.Slider(label="Width", minimum=256, maximum=MAX_IMAGE_SIZE, step=32, value=DEFAULT_WIDTH)
108
- height = gr.Slider(label="Height", minimum=256, maximum=MAX_IMAGE_SIZE, step=32, value=DEFAULT_HEIGHT)
109
 
110
  with gr.Column(elem_classes="examples-gallery"):
111
  gr.Markdown("### Gallery")
@@ -113,21 +105,14 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
113
  examples=examples,
114
  fn=generate_image,
115
  inputs=[prompt],
116
- outputs=[result, seed, latency],
117
  cache_examples="lazy"
118
  )
119
 
120
- enhanceBtn.click(
121
- fn=generate_image,
122
- inputs=[prompt, seed, width, height],
123
- outputs=[result, seed, latency],
124
- show_progress="full",
125
- )
126
-
127
  generateBtn.click(
128
  fn=generate_image,
129
  inputs=[prompt, seed, width, height, randomize_seed],
130
- outputs=[result, seed, latency],
131
  show_progress="full",
132
  api_name="GenerateImage",
133
  )
 
30
  seed = random.randint(0, MAX_SEED)
31
  generator = torch.Generator().manual_seed(int(float(seed)))
32
 
 
 
33
  img = pipe.generate_images(
34
  prompt=prompt,
35
  width=width,
 
37
  num_inference_steps=DEFAULT_INFERENCE_STEPS,
38
  generator=generator
39
  )
40
+ return img, seed
 
41
 
42
  # Example prompts
43
  examples = [
 
50
  "A stylish young woman with long, wavy ombre hair and winged eyeliner, posing in front of a neon-lit city skyline at night. She's wearing a sleek black leather jacket over a sparkly crop top and holding up a holographic business card that says 'Invite' in futuristic font. The card reflects the colorful neon lights, creating a cyberpunk aesthetic.",
51
  ]
52
 
 
53
  css = """
54
  footer {visibility: hidden;}
55
  .container {max-width: 1200px; margin: auto; padding: 20px;}
 
58
  border-radius: 10px;
59
  padding: 20px;
60
  margin-bottom: 20px;
61
+ height: 600px; /* 고정된 높이 설정 */
62
  display: flex;
63
  flex-direction: column;
64
  }
 
89
  placeholder="sexy woman & man , under wear, full body, sunday",
90
  lines=3,
91
  )
92
+ generateBtn = gr.Button("Generate Image", variant="primary")
 
 
 
 
93
 
94
+ with gr.Column(elem_classes="advanced-options"):
95
+ with gr.Row():
96
+ seed = gr.Number(label="Seed", value=42)
97
+ randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
98
+ with gr.Row():
99
+ width = gr.Slider(label="Width", minimum=256, maximum=MAX_IMAGE_SIZE, step=32, value=DEFAULT_WIDTH)
100
+ height = gr.Slider(label="Height", minimum=256, maximum=MAX_IMAGE_SIZE, step=32, value=DEFAULT_HEIGHT)
101
 
102
  with gr.Column(elem_classes="examples-gallery"):
103
  gr.Markdown("### Gallery")
 
105
  examples=examples,
106
  fn=generate_image,
107
  inputs=[prompt],
108
+ outputs=[result, seed],
109
  cache_examples="lazy"
110
  )
111
 
 
 
 
 
 
 
 
112
  generateBtn.click(
113
  fn=generate_image,
114
  inputs=[prompt, seed, width, height, randomize_seed],
115
+ outputs=[result, seed],
116
  show_progress="full",
117
  api_name="GenerateImage",
118
  )