aiqcamp commited on
Commit
43af73f
โ€ข
1 Parent(s): 4b8e8f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -129
app.py CHANGED
@@ -23,19 +23,6 @@ pipe.to("cuda")
23
 
24
  MAX_SEED = 2**32-1
25
 
26
- # ์˜ˆ์ œ ์„ค์ •
27
- example_model_prompts = [
28
- "professional fashion model, full body shot, standing pose, natural lighting, studio background, high fashion, elegant pose",
29
- "fashion model portrait, upper body, confident pose, fashion photography, neutral background, professional lighting",
30
- "stylish fashion model, three-quarter view, editorial pose, high-end fashion magazine style, minimal background"
31
- ]
32
-
33
- example_clothes_prompts = [
34
- "luxury designer sweater, cashmere material, cream color, cable knit pattern, high-end fashion, product photography",
35
- "elegant business blazer, tailored fit, charcoal grey, premium wool fabric, professional wear",
36
- "modern streetwear hoodie, oversized fit, minimalist design, premium cotton, urban style"
37
- ]
38
-
39
  @spaces.GPU()
40
  def generate_fashion(prompt, mode, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
41
  # ํ•œ๊ธ€ ๊ฐ์ง€ ๋ฐ ๋ฒˆ์—ญ
@@ -79,106 +66,9 @@ def generate_fashion(prompt, mode, cfg_scale, steps, randomize_seed, seed, width
79
  progress(100, "Completed!")
80
  return image, seed
81
 
82
- # CSS ์ •์˜
83
- custom_css = """
84
- body {
85
- background-color: #f5f5f5;
86
- font-family: 'Arial', sans-serif;
87
- }
88
-
89
- .container {
90
- max-width: 1200px;
91
- margin: 0 auto;
92
- padding: 20px;
93
- }
94
-
95
- .header {
96
- text-align: center;
97
- color: #333;
98
- margin-bottom: 30px;
99
- font-size: 2.5em;
100
- text-transform: uppercase;
101
- letter-spacing: 2px;
102
- }
103
-
104
- .box-common {
105
- background-color: white;
106
- border-radius: 15px;
107
- padding: 25px;
108
- margin: 20px 0;
109
- box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
110
- }
111
-
112
- .mode-box {
113
- background-color: #fff;
114
- padding: 20px;
115
- border-radius: 10px;
116
- margin-bottom: 20px;
117
- border: 2px solid #eee;
118
- }
119
-
120
- .result-box {
121
- width: 90%;
122
- max-width: 1000px;
123
- margin: 20px auto;
124
- }
125
-
126
- .image-output {
127
- width: 100%;
128
- max-width: 800px;
129
- margin: 0 auto;
130
- display: block;
131
- }
132
-
133
- .prompt-box {
134
- width: 90%;
135
- max-width: 1000px;
136
- margin: 20px auto;
137
- }
138
-
139
- .generate-btn {
140
- background: linear-gradient(45deg, #ff6b6b, #ff8e8e) !important;
141
- color: white !important;
142
- padding: 15px 30px !important;
143
- border-radius: 8px !important;
144
- border: none !important;
145
- font-size: 1.1em !important;
146
- cursor: pointer !important;
147
- transition: all 0.3s ease !important;
148
- width: 250px !important;
149
- margin: 20px auto !important;
150
- display: block !important;
151
- text-transform: uppercase !important;
152
- letter-spacing: 1px !important;
153
- }
154
-
155
- .generate-btn:hover {
156
- background: linear-gradient(45deg, #ff8e8e, #ff6b6b) !important;
157
- transform: translateY(-2px) !important;
158
- box-shadow: 0 5px 15px rgba(255, 107, 107, 0.4) !important;
159
- }
160
-
161
- .accordion {
162
- width: 90%;
163
- max-width: 1000px;
164
- margin: 20px auto;
165
- }
166
-
167
- .examples-table {
168
- background-color: rgba(255, 255, 255, 0.95);
169
- border-radius: 8px;
170
- padding: 15px;
171
- margin-top: 20px;
172
- }
173
-
174
- .parameter-box {
175
- background-color: #f8f9fa;
176
- padding: 20px;
177
- border-radius: 8px;
178
- margin: 10px 0;
179
- }
180
- """
181
-
182
 
183
  with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange") as app:
184
  gr.Markdown("# ๐ŸŽญ Fashion AI Studio")
@@ -219,17 +109,9 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange") as app:
219
  label="Style Transfer Strength"
220
  )
221
  tryon_btn = gr.Button("Generate Try-On")
222
-
223
- gr.Examples(
224
- examples=examples,
225
- inputs=[prompt_input, structure_image, style_image, depth_strength, style_strength],
226
- outputs=[output_image],
227
- fn=generate_image,
228
- cache_examples=True
229
- )
230
 
231
  with gr.Column():
232
- output_image.render()
233
 
234
  # Fashion Generation ํƒญ
235
  with gr.TabItem("๐Ÿ‘— Fashion Generation"):
@@ -269,13 +151,6 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange") as app:
269
  randomize_seed = gr.Checkbox(True, label="Randomize seed")
270
  seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, value=42)
271
 
272
- # ์˜ˆ์ œ ํƒญ
273
- with gr.Tabs():
274
- with gr.TabItem("Model Examples"):
275
- gr.Examples(examples=example_model_prompts, inputs=prompt)
276
- with gr.TabItem("Clothes Examples"):
277
- gr.Examples(examples=example_clothes_prompts, inputs=prompt)
278
-
279
  # ์ด๋ฒคํŠธ ํ•ธ๋“ค๋Ÿฌ
280
  tryon_btn.click(
281
  fn=generate_image,
 
23
 
24
  MAX_SEED = 2**32-1
25
 
 
 
 
 
 
 
 
 
 
 
 
 
 
26
  @spaces.GPU()
27
  def generate_fashion(prompt, mode, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
28
  # ํ•œ๊ธ€ ๊ฐ์ง€ ๋ฐ ๋ฒˆ์—ญ
 
66
  progress(100, "Completed!")
67
  return image, seed
68
 
69
+ def generate_image(prompt, structure_image, style_image, depth_strength, style_strength):
70
+ # ์‹ค์ œ ์ด๋ฏธ์ง€ ์ƒ์„ฑ ๋กœ์ง์„ ์—ฌ๊ธฐ์— ๊ตฌํ˜„
71
+ return Image.new('RGB', (512, 512), 'white')
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
72
 
73
  with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange") as app:
74
  gr.Markdown("# ๐ŸŽญ Fashion AI Studio")
 
109
  label="Style Transfer Strength"
110
  )
111
  tryon_btn = gr.Button("Generate Try-On")
 
 
 
 
 
 
 
 
112
 
113
  with gr.Column():
114
+ output_image = gr.Image(label="Generated Result")
115
 
116
  # Fashion Generation ํƒญ
117
  with gr.TabItem("๐Ÿ‘— Fashion Generation"):
 
151
  randomize_seed = gr.Checkbox(True, label="Randomize seed")
152
  seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, value=42)
153
 
 
 
 
 
 
 
 
154
  # ์ด๋ฒคํŠธ ํ•ธ๋“ค๋Ÿฌ
155
  tryon_btn.click(
156
  fn=generate_image,