Update app.py

app.py

@@ -16,7 +16,7 @@ if not os.path.exists(SAVE_DIR):
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 repo_id = "black-forest-labs/FLUX.1-dev"
-adapter_id = "openfree/van-gogh"  # Changed to van Gogh model
+adapter_id = "seawolf2357/kim-korea"  # 특정 정치인을 학습한 LoRA 모델
 
 pipeline = DiffusionPipeline.from_pretrained(repo_id, torch_dtype=torch.bfloat16)
 pipeline.load_lora_weights(adapter_id)
@@ -74,22 +74,22 @@ def inference(
     # Return just the image and seed
     return image, seed
 
-# Updated examples with 1880s clothing style for Van Gogh
+# 예시 문구: 특정 정치인 Mr. KIM의 다양한 상황을 묘사
 examples = [
-    "van Gogh's painting of a lively outdoor gathering in the 1880s, with men in formal top hats, frock coats, and women in bustled dresses with elaborate hats, enjoying a summer afternoon. The scene captures the Belle Époque atmosphere with swirling brushstrokes and vibrant colors typical of van Gogh, highlighting the fashionable attire of the period. [trigger]",
-    "van Gogh's intimate portrait of a young woman from the 1880s, with her hair styled in a fashionable updo, wearing a high-necked dress with lace details and leg-of-mutton sleeves. She wears delicate jewelry and has the expressive, bold brushstrokes and vibrant color palette characteristic of van Gogh's portraiture. [trigger]",
-    "van Gogh's painting of two young girls in 1880s attire seated at a piano. One plays while the other stands nearby, both dressed in white frocks with ribbon details, sashes, and high collars typical of the period. The interior setting features swirling patterns, intense colors, and the bold impasto technique typical of van Gogh's style. [trigger]",
-    "van Gogh's painting of an elegant 1880s boating party, with gentlemen in striped boating blazers, straw boater hats, and formal trousers, alongside ladies in bustled day dresses with parasols. The scene captures the leisure activities of French society during the Belle Époque era, rendered with the artist's characteristic swirling brushstrokes and vibrant contrasting colors. [trigger]",
-    "van Gogh's painting of children playing in an 1880s garden scene, dressed in formal period children's wear including sailor suits for boys and pinafores with full skirts for girls. Their Victorian-era clothing contrasts with their playful activities, set against van Gogh's vibrant treatment of nature with bold, expressive brushstrokes and dramatic color combinations. [trigger]",
-    "van Gogh's depiction of bathers in 1880s swimming attire, showing the modest bathing costumes of the period. Women wear full-coverage dark bathing dresses with stockings, while men are in knee-length swimming suits. The figures are arranged against a backdrop of water and landscape rendered in van Gogh's distinctive swirling brushwork and bold chromatic contrasts. [trigger]"
+    "Mr. KIM delivering a powerful speech in front of a large crowd with confident gestures and determined expression. [trigger]",
+    "Mr. KIM attending an international summit, standing among other world leaders with a dignified and poised manner. [trigger]",
+    "Mr. KIM holding a press conference, facing flashing cameras, wearing a tailored suit and a subtle lapel pin. [trigger]",
+    "Mr. KIM visiting a rural area, warmly greeting local residents while discussing policies and improvements. [trigger]",
+    "Mr. KIM in a dynamic interview setting, passionately outlining his visions for the future. [trigger]",
+    "Mr. KIM preparing for an important debate, surrounded by paperwork, looking focused and resolute. [trigger]",
 ]
 
-# Improved custom CSS with better visuals - updated colors for Van Gogh theme
+# UI를 붉은 계열 그라디에이션으로 디자인
 custom_css = """
 :root {
-    --color-primary: #1A3A8F;
-    --color-secondary: #E09900;
-    --background-fill-primary: linear-gradient(to right, #f6f9fc, #f0f4e3);
+    --color-primary: #8F1A3A; /* 붉은 톤의 메인 컬러 */
+    --color-secondary: #FF4B4B; /* 포인트 컬러(밝은 빨강) */
+    --background-fill-primary: linear-gradient(to right, #FFF5F5, #FED7D7, #FEB2B2);
 }
 
 footer {
@@ -146,6 +146,7 @@ footer {
 button.primary {
     background-color: var(--color-primary) !important;
     transition: all 0.3s ease;
+    color: #fff !important;
 }
 
 button:hover {
@@ -156,6 +157,9 @@ button:hover {
 .input-container {
     border-radius: 10px;
     box-shadow: 0 2px 8px rgba(0,0,0,0.05);
+    background-color: rgba(255, 255, 255, 0.6);
+    padding: 20px;
+    margin-bottom: 1rem;
 }
 
 .advanced-settings {
@@ -174,23 +178,29 @@ button:hover {
 """
 
 with gr.Blocks(css=custom_css, analytics_enabled=False) as demo:
-    gr.HTML('<div class="title">Vincent van Gogh STUDIO</div>')
+    gr.HTML('<div class="title">Mr. KIM in KOREA</div>')
     
-    # Add collection link below title
-    gr.HTML('<div class="collection-link"><a href="https://huggingface.co/collections/openfree/painting-art-ai-681453484ec15ef5978bbeb1" target="_blank">View the full Painting Art AI Collection</a></div>')
+    # 컬렉션 ���크 또는 안내문을 필요 시 수정/삭제
+    gr.HTML('<div class="collection-link"><a href="https://huggingface.co/collections/openfree/painting-art-ai-681453484ec15ef5978bbeb1" target="_blank">Visit the LoRA Model Collection</a></div>')
     
-    # Model description with the requested content
+    # 모델 설명: 특정 정치인에 대한 LoRA 모델임을 언급
     with gr.Group(elem_classes="model-description"):
-        gr.HTML('<p>Generate beautiful artwork in the style of Vincent van Gogh. Add [trigger] at the end of your prompt for best results.</p>')
+        gr.HTML("""
+        <p>
+        본 모델은 특정 정치인의 얼굴과 외모를 학습한 LoRA 모델입니다.<br>
+        개인 사진 등을 무단 사용하지 않도록 유의해 주세요.<br>
+        (예시 prompt 사용 시 [trigger]를 추가하여 최적의 결과를 얻을 수 있습니다.)
+        </p>
+        """)
 
-    # Simplified structure without tabs and gallery
+    # 메인 UI
     with gr.Column(elem_id="col-container"):
         with gr.Row(elem_classes="input-container"):
             prompt = gr.Text(
                 label="Prompt",
                 max_lines=1,
                 placeholder="Enter your prompt (add [trigger] at the end)",
-                value=examples[0]  # Set default text instead of generating an image
+                value=examples[0]  # 기본 예시
             )
             run_button = gr.Button("Generate", variant="primary", scale=0)
 
@@ -252,11 +262,11 @@ with gr.Blocks(css=custom_css, analytics_enabled=False) as demo:
                 examples=examples,
                 inputs=prompt,
                 outputs=None,  # Don't auto-run examples
-                fn=None,  # No function to run for examples - just fill the prompt
-                cache_examples=False,  # Disable caching
+                fn=None,       # No function to run for examples - just fill the prompt
+                cache_examples=False,
             )
 
-    # Event handlers
+    # 이벤트 핸들러
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=inference,
@@ -272,8 +282,6 @@ with gr.Blocks(css=custom_css, analytics_enabled=False) as demo:
         ],
         outputs=[result, seed_output],
     )
-    
-    # No preloading or automatic image generation
 
 demo.queue()
-demo.launch()
+demo.launch()