Update app.py

app.py

@@ -6,27 +6,25 @@ import random
 import numpy as np
 import os
 
-
+# GPU 사용 가능 여부 확인
 if torch.cuda.is_available():
     device = "cuda"
-    print("Using GPU")
+    print("GPU를 사용합니다")
 else:
     device = "cpu"
-    print("Using CPU")
-
+    print("CPU를 사용합니다")
 
-# login hf token
+# HuggingFace 토큰 로그인
 HF_TOKEN = os.getenv("HF_TOKEN")
 
-
 MAX_SEED = np.iinfo(np.int32).max
 CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "0") == "1"
 
-# Initialize the pipeline and download the model
+# 파이프라인 초기화 및 모델 다운로드
 pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
 pipe.to(device)
 
-# Define the image generation function
+# 이미지 생성 함수 정의
 @spaces.GPU(duration=160)
 def generate_image(prompt, num_inference_steps, height, width, guidance_scale, seed, num_images_per_prompt, progress=gr.Progress(track_tqdm=True)):
     if seed == 0:
@@ -34,7 +32,6 @@ def generate_image(prompt, num_inference_steps, height, width, guidance_scale, s
 
     generator = torch.Generator().manual_seed(seed)
     
-    
     with torch.inference_mode():
         output = pipe(
             prompt=prompt,
@@ -48,51 +45,140 @@ def generate_image(prompt, num_inference_steps, height, width, guidance_scale, s
     
     return output
 
-
-
-# Create the Gradio interface
-
+# 예제 프롬프트
 examples = [
-    ["A cat holding a sign that says hello world"],
-    ["a tiny astronaut hatching from an egg on the moon"],
-    ["An astrounat on mars in a futuristic cyborg suit."],
+    ["안녕하는 팻말을 들고 있는 고양이"],
+    ["달에서 알을 깨고 나오는 작은 우주인"],
+    ["미래적인 사이보그 슈트를 입고 화성에 있는 우주인"],
 ]
 
+# 커스텀 CSS
 css = '''
-.gradio-container{max-width: 1000px !important}
-h1{text-align:center}
+.gradio-container {
+    max-width: 1000px !important;
+    margin: auto;
+}
+h1 {
+    text-align: center;
+    font-family: 'Pretendard', sans-serif;
+    color: #EA580C;
+}
+.gr-button-primary {
+    background-color: #F97316 !important;
+}
+.gr-button-primary:hover {
+    background-color: #EA580C !important;
+}
 '''
-with gr.Blocks(css=css) as demo:
+
+# Gradio 인터페이스 생성
+with gr.Blocks(
+    theme=gr.themes.Soft(
+        primary_hue=gr.themes.Color(
+            c50="#FFF7ED",
+            c100="#FFEDD5",
+            c200="#FED7AA",
+            c300="#FDBA74",
+            c400="#FB923C",
+            c500="#F97316",
+            c600="#EA580C",
+            c700="#C2410C",
+            c800="#9A3412",
+            c900="#7C2D12",
+            c950="#431407",
+        ),
+        secondary_hue="zinc",
+        neutral_hue="zinc",
+        font=("Pretendard", "sans-serif")
+    ),
+    css=css
+) as demo:
     with gr.Row():
         with gr.Column():
             gr.HTML(
-            """
-            <h1 style='text-align: center'>
-            FLUX.1-dev
-            </h1>
-            """
-        )
+                """
+                <h1>FLUX.1-dev 이미지 생성기</h1>
+                """
+            )
             gr.HTML(
                 """
-               Made by <a href='https://linktr.ee/Nick088' target='_blank'>Nick088</a>
-               <br> <a href="https://discord.gg/AQsmBmgEPy"> <img src="https://img.shields.io/discord/1198701940511617164?color=%23738ADB&label=Discord&style=for-the-badge" alt="Discord"> </a>
+                <div style='text-align: center'>
+                제작: <a href='https://linktr.ee/Nick088' target='_blank'>Nick088</a>
+                <br> 
+                <a href="https://discord.gg/AQsmBmgEPy">
+                    <img src="https://img.shields.io/discord/1198701940511617164?color=%23738ADB&label=디스코드&style=for-the-badge" alt="Discord">
+                </a>
+                </div>
                 """
-        )
+            )
+    
     with gr.Group():
         with gr.Column():
-            prompt = gr.Textbox(label="Prompt", info="Describe the image you want", placeholder="A cat...")
-            run_button = gr.Button("Run")
-        result = gr.Gallery(label="Generated AI Images", elem_id="gallery")
-    with gr.Accordion("Advanced options", open=False):
+            prompt = gr.Textbox(
+                label="프롬프트",
+                info="원하는 이미지를 설명해주세요",
+                placeholder="고양이..."
+            )
+            run_button = gr.Button("생성하기", variant="primary")
+        result = gr.Gallery(
+            label="생성된 AI 이미지",
+            elem_id="gallery"
+        )
+    
+    with gr.Accordion("고급 설정", open=False):
         with gr.Row():
-            num_inference_steps = gr.Slider(label="Number of Inference Steps", info="The number of denoising steps of the image. More denoising steps usually lead to a higher quality image at the cost of slower inference", minimum=1, maximum=50, value=25, step=1)
-            guidance_scale = gr.Slider(label="Guidance Scale", info="Controls how much the image generation process follows the text prompt. Higher values make the image stick more closely to the input text.", minimum=0.0, maximum=7.0, value=3.5, step=0.1)
+            num_inference_steps = gr.Slider(
+                label="추론 단계 수",
+                info="이미지의 디노이징 단계 수입니다. 더 많은 단계는 더 높은 품질의 이미지를 생성하지만 시간이 더 걸립니다",
+                minimum=1,
+                maximum=50,
+                value=25,
+                step=1
+            )
+            guidance_scale = gr.Slider(
+                label="가이던스 스케일",
+                info="텍스트 프롬프트를 얼마나 충실히 따를지 제어합니다. 높은 값은 입력 텍스트에 더 가깝게 생성됩니다",
+                minimum=0.0,
+                maximum=7.0,
+                value=3.5,
+                step=0.1
+            )
+        
         with gr.Row():
-            width = gr.Slider(label="Width", info="Width of the Image", minimum=256, maximum=1024, step=32, value=1024)
-            height = gr.Slider(label="Height", info="Height of the Image", minimum=256, maximum=1024, step=32, value=1024)
+            width = gr.Slider(
+                label="너비",
+                info="이미지의 너비",
+                minimum=256,
+                maximum=1024,
+                step=32,
+                value=1024
+            )
+            height = gr.Slider(
+                label="높이",
+                info="이미지의 높이",
+                minimum=256,
+                maximum=1024,
+                step=32,
+                value=1024
+            )
+        
         with gr.Row():
-            seed = gr.Slider(value=42, minimum=0, maximum=MAX_SEED, step=1, label="Seed", info="A starting point to initiate the generation process, put 0 for a random one")
-            num_images_per_prompt = gr.Slider(label="Images Per Prompt", info="Number of Images to generate with the settings",minimum=1, maximum=4, step=1, value=2)
+            seed = gr.Slider(
+                value=42,
+                minimum=0,
+                maximum=MAX_SEED,
+                step=1,
+                label="시드",
+                info="생성 과정의 시작점입니다. 0을 입력하면 랜덤한 시드가 사용됩니다"
+            )
+            num_images_per_prompt = gr.Slider(
+                label="프롬프트당 이미지 수",
+                info="설정된 값으로 생성할 이미지의 수",
+                minimum=1,
+                maximum=4,
+                step=1,
+                value=2
+            )
 
     gr.Examples(
         examples=examples,