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/korea-president-yoon"
+adapter_id = "openfree/pepe"
 
 pipeline = DiffusionPipeline.from_pretrained(repo_id, torch_dtype=torch.bfloat16)
 pipeline.load_lora_weights(adapter_id)
@@ -47,8 +47,11 @@ def load_generated_images():
         return []
     
     # Load all images from the directory
-    image_files = [os.path.join(SAVE_DIR, f) for f in os.listdir(SAVE_DIR) 
-                  if f.endswith(('.png', '.jpg', '.jpeg', '.webp'))]
+    image_files = [
+        os.path.join(SAVE_DIR, f)
+        for f in os.listdir(SAVE_DIR)
+        if f.endswith(('.png', '.jpg', '.jpeg', '.webp'))
+    ]
     # Sort by creation time (newest first)
     image_files.sort(key=lambda x: os.path.getctime(x), reverse=True)
     return image_files
@@ -57,6 +60,19 @@ def load_predefined_images():
     # Return empty list since we're not using predefined images
     return []
 
+# 간단한 CSS: 타이틀 크기, 중앙 정렬, footer 숨김 등
+css = """
+.title {
+    font-size: 1.8em;
+    font-weight: bold;
+    text-align: center;
+    margin: 20px 0;
+}
+footer {
+    visibility: hidden;
+}
+"""
+
 @spaces.GPU(duration=120)
 def inference(
     prompt: str,
@@ -90,35 +106,33 @@ def inference(
     return image, seed, load_generated_images()
 
 examples = [
-    "A man playing fetch with a golden retriever in a sunny park. He wears casual weekend clothes and throws a red frisbee with joy. The dog leaps gracefully through the air, tail wagging with excitement. Warm afternoon sunlight filters through the trees, creating a peaceful scene of companionship. [president yoon]",
+    "Pepe the frog playing fetch with a golden retriever in a sunny park. He wears casual weekend clothes and tosses a bright red frisbee with a goofy grin. The dog leaps gracefully through the air, tail wagging with excitement. The warm afternoon sunlight filters through the trees, creating a humorous meme-like atmosphere. [pepe]",
 
-    "A soldier standing at attention in full military gear, holding a standard-issue rifle. His uniform is crisp and properly adorned with medals. Behind him, other soldiers march in formation during a military parade. The scene conveys discipline and duty. [president yoon]",
+    "Pepe the frog dressed in full military gear, standing at attention with a standard-issue rifle. His crisp uniform is adorned with cartoonish medals. Other frog soldiers march in formation behind him during a grand meme parade, conveying discipline mixed with comical charm. [pepe]",
 
-    "A medieval knight in gleaming armor, holding an ornate sword and shield. He stands proudly in front of a majestic castle, his cape flowing in the wind. The shield bears intricate heraldic designs, and sunlight glints off his polished armor. [president yoon]",
+    "A medieval Pepe knight in gleaming armor, proudly holding an ornate sword and shield. He stands in front of a majestic castle with a swirling moat. His shield features a cartoon frog crest, and sunlight gleams off his polished armor, adding a humorous yet epic feel. [pepe]",
 
-    "A charismatic political leader addressing a crowd from a podium. He wears a well-fitted suit and gestures confidently while speaking. The audience fills a large plaza, holding supportive banners and signs. News cameras capture the moment as he delivers his speech. [president yoon]",
+    "A charismatic Pepe the frog addressing a crowd from a podium. He wears a well-fitted suit and gestures with exaggerated cartoon expressions while speaking. The audience is filled with fellow frog characters holding supportive banners. Cameras capture this grand meme moment. [pepe]",
 
-    "A man enjoying a peaceful morning at home, reading a newspaper at his breakfast table. He wears comfortable home clothes and sips coffee from a favorite mug. Sunlight streams through the kitchen window, and a house plant adds a touch of nature to the cozy domestic scene. [president yoon]",
+    "Pepe the frog enjoying a peaceful morning at home, reading a newspaper at his kitchen table. He wears comfy pajamas and sips coffee from a novelty frog mug. Sunlight streams through the window, illuminating a quaint plant on the countertop in this cozy, meme-inspired scene. [pepe]",
 
-    "A businessman walking confidently through a modern office building. He carries a leather briefcase and wears a tailored navy suit. Floor-to-ceiling windows reveal a cityscape behind him, and his expression shows determination and purpose. [president yoon]"
+    "Businessman Pepe walking confidently through a sleek office lobby, briefcase in hand. He wears a tailored navy suit, and his wide frog eyes convey determination. Floor-to-ceiling windows reveal a bustling cityscape behind him, blending corporate professionalism with meme humor. [pepe]"
 ]
 
-css = """
-footer {
-    visibility: hidden;
-}
-"""
 
-with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, analytics_enabled=False) as demo:
-    gr.HTML('<div class="title"> President Yoon in KOREA </div>')
+# Blocks에서 기존 theme 제거 후 css만 적용
+with gr.Blocks(css=css, analytics_enabled=False) as demo:
+    gr.HTML('<div class="title"> PEPE Meme Generator </div>')
 
-    gr.HTML("""<a href="https://visitorbadge.io/status?path=https%3A%2F%2Fopenfree-korea-president-yoon.hf.space">
-               <img src="https://api.visitorbadge.io/api/visitors?path=https%3A%2F%2Fopenfree-korea-president-yoon.hf.space&countColor=%23263759" />
-               </a>""")
+    gr.HTML("""
+        <a href="https://visitorbadge.io/status?path=https%3A%2F%2Fopenfree-pepe.hf.space">
+            <img src="https://api.visitorbadge.io/api/visitors?path=https%3A%2F%2Fopenfree-pepe.hf.space&countColor=%23263759" />
+        </a>
+    """)
     
     with gr.Tabs() as tabs:
         with gr.Tab("Generation"):
-            with gr.Column(elem_id="col-container"):
+            with gr.Column():
                 with gr.Row():
                     prompt = gr.Text(
                         label="Prompt",
@@ -226,4 +240,4 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, analytics_enabled=Fa
     )
 
 demo.queue()
-demo.launch()
+demo.launch()