itayitay123 commited on
Commit
a75b08d
·
verified ·
1 Parent(s): a736c4b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -12
app.py CHANGED
@@ -69,7 +69,7 @@ def generate_outfits(input_image, n=1):
69
  generated_images = []
70
 
71
  for _ in range(n):
72
- result = pipe(prompt=prompt, image=init_image, strength=0.4, guidance_scale=7.5)
73
  generated_images.append(result.images[0])
74
 
75
  return generated_images
@@ -127,13 +127,11 @@ with gr.Blocks() as demo:
127
  generate_btn = gr.Button("Generate Recommendations")
128
 
129
  with gr.Row():
130
- output1 = gr.Image(label="Your Input")
131
- output2 = gr.Image(label="Similar Item 1")
132
- output3 = gr.Image(label="Similar Item 2")
133
- output4 = gr.Image(label="Similar Item 3")
134
- output5 = gr.Image(label="AI-Generated Outfit")
135
-
136
-
137
 
138
  examples = gr.Examples(
139
  examples=example_paths,
@@ -146,7 +144,4 @@ with gr.Blocks() as demo:
146
  outputs=[output1, output2, output3, output4, output5])
147
 
148
  if __name__ == "__main__":
149
- demo.launch()
150
-
151
-
152
-
 
69
  generated_images = []
70
 
71
  for _ in range(n):
72
+ result = pipe(prompt=prompt, image=init_image, strength=0.7, guidance_scale=7.5)
73
  generated_images.append(result.images[0])
74
 
75
  return generated_images
 
127
  generate_btn = gr.Button("Generate Recommendations")
128
 
129
  with gr.Row():
130
+ output1 = gr.Image(label="Your Input", height=512, width=384)
131
+ output2 = gr.Image(label="Similar Item 1", height=512, width=384)
132
+ output3 = gr.Image(label="Similar Item 2", height=512, width=384)
133
+ output4 = gr.Image(label="Similar Item 3", height=512, width=384)
134
+ output5 = gr.Image(label="AI-Generated Outfit", height=512, width=384)
 
 
135
 
136
  examples = gr.Examples(
137
  examples=example_paths,
 
144
  outputs=[output1, output2, output3, output4, output5])
145
 
146
  if __name__ == "__main__":
147
+ demo.launch()