Update app.py

app.py

@@ -16,7 +16,7 @@ from huggingface_hub import snapshot_download
 import spaces 
 
 device = "cuda"
-root_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+root_dir = os.getcwd()
 ckpt_dir = f'{root_dir}/weights/Kolors'
 
 snapshot_download(repo_id="Kwai-Kolors/Kolors", local_dir=ckpt_dir)
@@ -48,9 +48,6 @@ pipe = StableDiffusionXLPipeline(
     force_zeros_for_empty_prompt=False
 ).to(device)
 
-#pipe = pipe.to(device)
-#pipe.enable_model_cpu_offload()
-
 if hasattr(pipe.unet, 'encoder_hid_proj'):
     pipe.unet.text_encoder_hid_proj = pipe.unet.encoder_hid_proj
 
@@ -85,9 +82,9 @@ def infer(prompt, ip_adapter_image, ip_adapter_scale=0.5, negative_prompt="", se
     return image, seed
 
 examples = [
-    ["A dog", "minta.jpeg", 0.4],
-    ["A capybara", "king-min.png", 0.5],
-    ["A cat", "blue_hair.png", 0.5],
+    ["강아지", "minta.jpeg", 0.4],
+    ["카피바라라", "king-min.png", 0.5],
+    ["고양이", "blue_hair.png", 0.5],
     ["", "meow.jpeg", 1.0],
 ]
 
@@ -107,56 +104,56 @@ css="""
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(f"""
-        # Kolors IP-Adapter - image reference and variations
+        # Kolors IP-Adapter - 이미지 참조 및 변형
         """)
-        
+
         with gr.Row():
             prompt = gr.Text(
-                label="Prompt",
+                label="프롬프트",
                 show_label=False,
                 max_lines=1,
-                placeholder="Enter your prompt",
+                placeholder="프롬프트를 입력하세요",
                 container=False,
             )
-            run_button = gr.Button("Run", scale=0)
-        
+            run_button = gr.Button("실행", scale=0)
+
         with gr.Row():
             with gr.Column():
-                ip_adapter_image = gr.Image(label="IP-Adapter Image", type="pil")
+                ip_adapter_image = gr.Image(label="IP-어댑터 이미지", type="pil")
                 ip_adapter_scale = gr.Slider(
-                    label="Image influence scale",
-                    info="Use 1 for creating variations",
+                    label="이미지 영향 척도",
+                    info="변형을 생성하려면 1을 사용하세요",
                     minimum=0.0,
                     maximum=1.0,
                     step=0.05,
                     value=0.5,
                 )
-            result = gr.Image(label="Result", elem_id="result")
+            result = gr.Image(label="결과", elem_id="result")
 
-        with gr.Accordion("Advanced Settings", open=False):
+        with gr.Accordion("고급 설정", open=False):
             negative_prompt = gr.Text(
-                label="Negative prompt",
+                label="부정적 프롬프트",
                 max_lines=1,
-                placeholder="Enter a negative prompt",
+                placeholder="부정적 프롬프트를 입력하세요",
             )
             seed = gr.Slider(
-                label="Seed",
+                label="시드",
                 minimum=0,
                 maximum=MAX_SEED,
                 step=1,
                 value=0,
             )
-            randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
+            randomize_seed = gr.Checkbox(label="시드 무작위화", value=True)
             with gr.Row():
                 width = gr.Slider(
-                    label="Width",
+                    label="너비",
                     minimum=256,
                     maximum=MAX_IMAGE_SIZE,
                     step=32,
                     value=1024,
                 )
                 height = gr.Slider(
-                    label="Height",
+                    label="높이",
                     minimum=256,
                     maximum=MAX_IMAGE_SIZE,
                     step=32,
@@ -164,18 +161,18 @@ with gr.Blocks(css=css) as demo:
                 )
             with gr.Row():
                 guidance_scale = gr.Slider(
-                    label="Guidance scale",
+                    label="가이던스 척도",
                     minimum=0.0,
                     maximum=10.0,
                     step=0.1,
                     value=5.0,
                 )
                 num_inference_steps = gr.Slider(
-                    label="Number of inference steps",
+                    label="추론 단계 수",
                     minimum=1,
                     maximum=100,
                     step=1,
-                    value=100,
+                    value=50,
                 )
             
         gr.Examples(
@@ -193,6 +190,5 @@ with gr.Blocks(css=css) as demo:
         outputs=[result, seed]
     )
 
-# 포트 7890 설정, 대기열 활성화, API 활성화
-demo.launch()
-
+# Launch the app
+demo.launch(share=True)