Update app.py

app.py

@@ -157,8 +157,8 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
                 offload: bool = False):
     
     with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
-        gr.Markdown("# AI 포토 지니: 우주")
-        gr.Markdown("## 이용 안내: 1) 예시중 하나를 선택. 2) 카메라 버튼을 클릭하고 얼굴이 보이면 카메라 버튼 클릭. 3) '생성' 버튼을 클릭하고 기다리면 됩니다.")
+        gr.Markdown("## AI 포토 지니: 우주")
+        gr.Markdown("### 이용 안내: 1) 예시중 하나를 선택. 2) 카메라 버튼을 클릭하고 얼굴이 보이면 카메라 버튼 클릭. 3) '생성' 버튼을 클릭하고 기다리면 됩니다.")
         
         with gr.Row():
             with gr.Column():
@@ -171,7 +171,7 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
 
         with gr.Row():
             with gr.Column():
-                gr.Markdown("## 예시")
+                gr.Markdown("### 예시")
                 
                 all_examples = [
                     ['여자가 "PuLID for FLUX"라고 쓰인 빛나는 녹색 표지판을 들고 있다', 'example_inputs/liuyifei.png'],
@@ -210,7 +210,7 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
                 )
 
         generate_btn.click(
-            fn=generate_image,
+            fn=lambda *args: generate_image(*args)[0],  # Only return the first item (the image)
             inputs=[
                 gr.Slider(256, 1536, 896, step=16, visible=False),  # width
                 gr.Slider(256, 1536, 1152, step=16, visible=False),  # height
@@ -249,7 +249,4 @@ if __name__ == "__main__":
     huggingface_hub.login(os.getenv('HF_TOKEN'))
 
     demo = create_demo(args, args.name, args.device, args.offload)
-    demo.launch()
-
-
-
+    demo.launch(share=True)