Update app.py

app.py

@@ -153,6 +153,7 @@ footer {
 }
 """
 
+
 def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_available() else "cpu",
                 offload: bool = False):
     
@@ -173,27 +174,21 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
                 gr.Markdown("### 스타일")
                 
                 all_examples = [
-#우주 여행I            ['I am an astronaut on a spacewalk. There is no helmet, and my face is visible. The background is Earth & starship as seen from space shuttle.', 'example_inputs/1.webp'],
-#우주 여행II           ['I am an astronaut on a spacewalk. There is no helmet, and my face is visible. The background is Earth & starship as seen from space shuttle.I am holding sign with glowing green text "I Love Mom"', 'example_inputs/2.webp'],
-#내가 어른이 되면       ['profile photo of a 40-year-old Adult Looking straight ahead, wear suite', 'example_inputs/3.webp'],
-#아이언맨 변신          ['I am an "IRON MAN"', 'example_inputs/4.webp'],
-#화성 탐험             ['I am wearing a spacesuit and have become an astronaut walking on Mars. I'm not wearing a helmet. I'm looking straight ahead. The background is a desolate area of Mars, and a space rover and a space station can be seen.', 'example_inputs/5.webp'],                    
-#스파이더맨             ['I am an "spider MAN"', 'example_inputs/6.webp'],                    
-#우주선 조종            ['I am wearing a spacesuit and have become an astronaut. I am piloting a spacecraft. Through the spacecraft's window, I can see outer space.', 'example_inputs/7.webp'],
-#만화 주인공            ['portrait, pixar style', 'example_inputs/8.webp'],
-#원더우먼               ['I am an "wonder woman"', 'example_inputs/9.webp'],
-#카우보이               ['Cowboy, american comics style', 'example_inputs/10.webp'],
-
-
+                    ["우주 여행I", 'I am an astronaut on a spacewalk. There is no helmet, and my face is visible. The background is Earth & starship as seen from space shuttle.', 'example_inputs/1.webp'],
+                    ["우주 여행II", 'I am an astronaut on a spacewalk. There is no helmet, and my face is visible. The background is Earth & starship as seen from space shuttle.I am holding sign with glowing green text "I Love Mom"', 'example_inputs/2.webp'],
+                    ["우주선 조종", 'I am wearing a spacesuit and have become an astronaut. I am piloting a spacecraft. Through the spacecraft's window, I can see outer space.', 'example_inputs/7.webp'],
+                    ["화성 탐험", 'I am wearing a spacesuit and have become an astronaut walking on Mars. I'm not wearing a helmet. I'm looking straight ahead. The background is a desolate area of Mars, and a space rover and a space station can be seen.', 'example_inputs/5.webp'],
+                    ["아이언맨 변신", 'I am an "IRON MAN"', 'example_inputs/4.webp'],
+                    ["스파이더맨", 'I am an "spider MAN"', 'example_inputs/6.webp'],
+                    ["만화 주인공", 'portrait, pixar style', 'example_inputs/8.webp'],
+                    ["원더우먼", 'I am an "wonder woman"', 'example_inputs/9.webp'],
+                    ["카우보이", 'Cowboy, american comics style', 'example_inputs/10.webp'],
+                    ["내가 어른이 되면", 'profile photo of a 40-year-old Adult Looking straight ahead, wear suite', 'example_inputs/3.webp'],                    
                 ]
 
-                example_images = [example[1] for example in all_examples]
-                example_captions = [example[0] for example in all_examples]
-
-                gallery = gr.Gallery(
-                    value=list(zip(example_images, example_captions)),
-                    label="예시 갤러리",
-                    show_label=False,
+                example_gallery = gr.Gallery(
+                    [example[2] for example in all_examples],
+                    label="스타일 예시",
                     elem_id="gallery",
                     columns=5,
                     rows=2,
@@ -202,16 +197,16 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
                 )
 
                 def fill_example(evt: gr.SelectData):
-                    return [all_examples[evt.index][i] for i in [0, 1]]
+                    return all_examples[evt.index][1]
 
-                gallery.select(
+                example_gallery.select(
                     fill_example,
                     None,
-                    [prompt, id_image],
+                    [prompt],
                 )
 
         generate_btn.click(
-            fn=lambda *args: generate_image(*args)[0],  # Only return the first item (the image)
+            fn=lambda *args: generate_image(*args)[0],  # 첫 번째 항목(이미지)만 반환
             inputs=[
                 gr.Slider(256, 1536, 896, step=16, visible=False),  # width
                 gr.Slider(256, 1536, 1152, step=16, visible=False),  # height
@@ -232,6 +227,7 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
 
     return demo
 
+
 if __name__ == "__main__":
     import argparse