Update app.py

app.py

@@ -163,14 +163,12 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
  
  with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
  
- 
  with gr.Row():
  with gr.Column():
  prompt = gr.Textbox(label="Prompt", value="portrait, color, cinematic")
  id_image = gr.Image(label="ID Image")
  id_weight = gr.Slider(0.0, 3.0, 1, step=0.05, label="id weight")
 
-
  width = gr.Slider(256, 1536, 896, step=16, label="Width")
  height = gr.Slider(256, 1536, 1152, step=16, label="Height")
  num_steps = gr.Slider(1, 20, 20, step=1, label="Number of steps")
@@ -193,70 +191,51 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
  output_image = gr.Image(label="Generated Image")
  seed_output = gr.Textbox(label="Used Seed")
  intermediate_output = gr.Gallery(label='Output', elem_id="gallery", visible=args.dev)
- # _CITE_ 관련 부분 제거
- 
 
- with gr.Row(), gr.Column():
+ with gr.Row():
+ with gr.Column():
  gr.Markdown("## Examples")
- example_inps = [
- [
- 'a woman holding sign with glowing green text \"PuLID for FLUX\"',
- 'example_inputs/liuyifei.png',
- 4, 4, 2680261499100305976, 1
- ],
- [
- 'portrait, side view',
- 'example_inputs/liuyifei.png',
- 4, 4, 1205240166692517553, 1
- ],
- [
- 'white-haired woman with vr technology atmosphere, revolutionary exceptional magnum with remarkable details', # noqa E501
- 'example_inputs/liuyifei.png',
- 4, 4, 6349424134217931066, 1
- ],
- [
- 'a young child is eating Icecream',
- 'example_inputs/liuyifei.png',
- 4, 4, 10606046113565776207, 1
- ],
- [
- 'a man is holding a sign with text \"PuLID for FLUX\", winter, snowing, top of the mountain',
- 'example_inputs/pengwei.jpg',
- 4, 4, 2410129802683836089, 1
- ],
- [
- 'portrait, candle light',
- 'example_inputs/pengwei.jpg',
- 4, 4, 17522759474323955700, 1
- ],
- [
- 'profile shot dark photo of a 25-year-old male with smoke escaping from his mouth, the backlit smoke gives the image an ephemeral quality, natural face, natural eyebrows, natural skin texture, award winning photo, highly detailed face, atmospheric lighting, film grain, monochrome', # noqa E501
- 'example_inputs/pengwei.jpg',
- 4, 4, 17733156847328193625, 1
- ],
- [
- 'American Comics, 1boy',
- 'example_inputs/pengwei.jpg',
- 1, 4, 13223174453874179686, 1
- ],
- [
- 'portrait, pixar',
- 'example_inputs/pengwei.jpg',
- 1, 4, 9445036702517583939, 1
- ],
- ]
- gr.Examples(examples=example_inps, inputs=[prompt, id_image, start_step, guidance, seed, true_cfg],
- label='fake CFG')
-
- example_inps = [
- [
- 'portrait, made of ice sculpture',
- 'example_inputs/lecun.jpg',
- 1, 1, 3811899118709451814, 5
- ],
+ 
+ # 모든 예시를 하나의 리스트로 합칩니다
+ all_examples = [
+ ['a woman holding sign with glowing green text \"PuLID for FLUX\"', 'example_inputs/liuyifei.png', 4, 4, 2680261499100305976, 1],
+ ['portrait, side view', 'example_inputs/liuyifei.png', 4, 4, 1205240166692517553, 1],
+ ['white-haired woman with vr technology atmosphere, revolutionary exceptional magnum with remarkable details', 'example_inputs/liuyifei.png', 4, 4, 6349424134217931066, 1],
+ ['a young child is eating Icecream', 'example_inputs/liuyifei.png', 4, 4, 10606046113565776207, 1],
+ ['a man is holding a sign with text \"PuLID for FLUX\", winter, snowing, top of the mountain', 'example_inputs/pengwei.jpg', 4, 4, 2410129802683836089, 1],
+ ['portrait, candle light', 'example_inputs/pengwei.jpg', 4, 4, 17522759474323955700, 1],
+ ['profile shot dark photo of a 25-year-old male with smoke escaping from his mouth, the backlit smoke gives the image an ephemeral quality, natural face, natural eyebrows, natural skin texture, award winning photo, highly detailed face, atmospheric lighting, film grain, monochrome', 'example_inputs/pengwei.jpg', 4, 4, 17733156847328193625, 1],
+ ['American Comics, 1boy', 'example_inputs/pengwei.jpg', 1, 4, 13223174453874179686, 1],
+ ['portrait, pixar', 'example_inputs/pengwei.jpg', 1, 4, 9445036702517583939, 1],
+ ['portrait, made of ice sculpture', 'example_inputs/lecun.jpg', 1, 1, 3811899118709451814, 5],
  ]
- gr.Examples(examples=example_inps, inputs=[prompt, id_image, start_step, guidance, seed, true_cfg],
- label='true CFG')
+
+ # 예시 이미지와 캡션을 준비합니다
+ example_images = [example[1] for example in all_examples]
+ example_captions = [example[0] for example in all_examples]
+
+ # Gallery 컴포넌트를 사용하여 그리드 형태로 예시들을 표시합니다
+ gallery = gr.Gallery(
+ label="Example Gallery",
+ show_label=False,
+ elem_id="gallery",
+ columns=5,
+ rows=2,
+ object_fit="contain",
+ height="auto"
+ ).style(grid=5, height="auto")
+ 
+ gallery.value = list(zip(example_images, example_captions))
+
+ # 예시 클릭 시 입력 필드에 값을 채우는 함수
+ def fill_example(evt: gr.SelectData):
+ return [all_examples[evt.index][i] for i in [0, 1, 2, 3, 4, 5]]
+
+ gallery.select(
+ fill_example,
+ None,
+ [prompt, id_image, start_step, guidance, seed, true_cfg],
+ )
 
  generate_btn.click(
  fn=generate_image,