blanchon commited on
Commit
c11f3e7
·
1 Parent(s): d5aa8c4

Improve examples

Browse files
Files changed (1) hide show
  1. app.py +27 -5
app.py CHANGED
@@ -248,11 +248,13 @@ with gr.Blocks(css=css) as demo:
248
  image_mode="RGB",
249
  height=500,
250
  )
251
- furniture_prompt = gr.Text(
252
- label="Prompt",
253
- max_lines=1,
254
- placeholder="Enter a custom furniture description (optional)",
255
- container=False,
 
 
256
  )
257
  with gr.Column(elem_id="col-mid"):
258
  gr.HTML(
@@ -275,6 +277,19 @@ with gr.Blocks(css=css) as demo:
275
  brush=gr.Brush(colors=["#FFFFFF"], color_mode="fixed"),
276
  height=500,
277
  )
 
 
 
 
 
 
 
 
 
 
 
 
 
278
  with gr.Column(elem_id="col-right"):
279
  gr.HTML(
280
  """
@@ -303,6 +318,13 @@ with gr.Blocks(css=css) as demo:
303
  value=0,
304
  )
305
  randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
 
 
 
 
 
 
 
306
  with gr.Column():
307
  max_dimension = gr.Slider(
308
  label="Max Dimension",
 
248
  image_mode="RGB",
249
  height=500,
250
  )
251
+ furniture_examples = gr.Examples(
252
+ examples=[
253
+ EXAMPLES_DIR / "1" / "furniture_image.png",
254
+ EXAMPLES_DIR / "2" / "furniture_image.png",
255
+ ],
256
+ examples_per_page=12,
257
+ inputs=[furniture_image],
258
  )
259
  with gr.Column(elem_id="col-mid"):
260
  gr.HTML(
 
277
  brush=gr.Brush(colors=["#FFFFFF"], color_mode="fixed"),
278
  height=500,
279
  )
280
+ room_examples = gr.Examples(
281
+ examples=[
282
+ make_example(
283
+ EXAMPLES_DIR / "1" / "room_image.png",
284
+ EXAMPLES_DIR / "1" / "room_mask.png",
285
+ ),
286
+ make_example(
287
+ EXAMPLES_DIR / "2" / "room_image.png",
288
+ EXAMPLES_DIR / "2" / "room_mask.png",
289
+ ),
290
+ ],
291
+ inputs=[room_image],
292
+ )
293
  with gr.Column(elem_id="col-right"):
294
  gr.HTML(
295
  """
 
318
  value=0,
319
  )
320
  randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
321
+
322
+ furniture_prompt = gr.Text(
323
+ label="Prompt",
324
+ max_lines=1,
325
+ placeholder="Enter a custom furniture description (optional)",
326
+ container=False,
327
+ )
328
  with gr.Column():
329
  max_dimension = gr.Slider(
330
  label="Max Dimension",