itayitay123 commited on
Commit
8e9fd18
·
verified ·
1 Parent(s): 40e0c7c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -6
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.7, guidance_scale=7.5)
73
  generated_images.append(result.images[0])
74
 
75
  return generated_images
@@ -127,11 +127,12 @@ with gr.Blocks() as demo:
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,
 
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
  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
  examples = gr.Examples(
138
  examples=example_paths,