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/claude-monet"
+adapter_id = "openfree/pierre-auguste-renoir"  # Changed to Renoir model
 
 pipeline = DiffusionPipeline.from_pretrained(repo_id, torch_dtype=torch.bfloat16)
 pipeline.load_lora_weights(adapter_id)
@@ -55,12 +55,12 @@ def load_generated_images():
 
 def load_predefined_images():
     predefined_images = [
-        "assets/cm1.webp",
-        "assets/cm2.webp",
-        "assets/cm3.webp",
-        "assets/cm4.webp",
-        "assets/cm5.webp",
-        "assets/cm6.webp",
+        "assets/renoir1.webp",
+        "assets/renoir2.webp",
+        "assets/renoir3.webp",
+        "assets/renoir4.webp",
+        "assets/renoir5.webp",
+        "assets/renoir6.webp",
     ]
     return predefined_images
 
@@ -97,25 +97,93 @@ def inference(
     return image, seed, load_generated_images()
 
 examples = [
-    "Claude Monet's 1916 painting, Water Lilies, which is currently on display at the Metropolitan Museum of Art. The painting depicts a tranquil pond with water lilies floating on the surface, surrounded by lush green foliage and a variety of colorful flowers. The colors of the flowers range from bright pinks and purples to deep blues and greens, creating a peaceful and calming atmosphere. [trigger]",
-    "Claude Monet's 1869 masterpiece, The Magpie, showcasing a snow-covered rural landscape at dawn. A single black magpie perches on a wooden gate, contrasting against the pristine white snow. The scene captures the subtle interplay of light and shadow on the snow's surface, with delicate blue-gray tones in the shadows and warm golden hints where sunlight touches the snow-laden branches. [trigger]",
-    "Claude Monet's Impression, Sunrise (1872), depicting the port of Le Havre at dawn. The orange sun hangs low in a misty gray-blue sky, its reflection dancing across the rippling harbor waters. Small boats appear as dark silhouettes against the luminous morning light, while industrial chimneys in the background release wisps of smoke into the atmospheric scene. [trigger]",
-    "Claude Monet's Rouen Cathedral series (1892-1894), focusing on the western facade at sunset. The gothic architecture is bathed in warm golden light, with deep purple shadows in the intricate stone carvings. The cathedral's spires reach toward a sky painted in soft pinks and lavenders, showcasing Monet's masterful handling of light and atmospheric effects. [trigger]",
-    "Claude Monet's Japanese Bridge at Giverny (1899), featuring the iconic green curved bridge spanning his water garden. Clusters of purple and white wisteria cascade from above, their reflections merging with the lily pads below in the tranquil pond. Weeping willows frame the scene in characteristic Monet brushstrokes, creating a dreamy, impressionist atmosphere. [trigger]",
-    "Claude Monet's Haystacks at Sunset (1890), showing golden wheat stacks in a field at dusk. The massive forms of the haystacks stand silhouetted against a dramatic sky painted in bold strokes of orange, pink, and deep purple. The surrounding field catches the last rays of sunlight, creating a patchwork of warm earth tones and long blue-violet shadows. [trigger]"
+    "Renoir's painting of a lively outdoor dance scene at Moulin de la Galette, with dappled sunlight filtering through trees, illuminating well-dressed Parisians enjoying a summer afternoon. Couples dance while others socialize at tables, capturing the joie de vivre of 1870s Montmartre. [trigger]",
+    "Renoir's intimate portrait of a young woman with rosy cheeks and lips, soft blonde hair, and a gentle smile. She wears a vibrant blue dress against a background of lush flowers and greenery, showcasing his mastery of depicting feminine beauty with warm, luminous skin tones. [trigger]",
+    "Renoir's painting of two young girls seated at a piano, captured in his distinctive soft focus style. The scene shows one girl playing while the other stands beside her, both wearing delicate white dresses. The interior setting features warm colors and loose brushwork typical of his mature period. [trigger]",
+    "Renoir's painting of an elegant boating party, with fashionably dressed men and women relaxing on a restaurant terrace overlooking the Seine. The scene captures the leisurely atmosphere of 1880s French society, with sparkling water reflections and a bright, airy palette of blues, whites, and warm flesh tones. [trigger]",
+    "Renoir's painting of a sun-dappled garden scene with children playing. The composition features vibrant flowers in full bloom, lush greenery, and Renoir's characteristic luminous treatment of sunlight filtering through foliage, creating patches of brilliant color across the canvas. [trigger]",
+    "Renoir's depiction of bathers by a riverbank, with several female figures arranged in a harmonious composition. The painting showcases his later style with fuller figures rendered in pearlescent flesh tones against a backdrop of shimmering water and verdant landscape, demonstrating his unique approach to the nude figure in nature. [trigger]"
 ]
 
-css = """
+# Brighter custom CSS with vibrant colors
+custom_css = """
+:root {
+    --color-primary: #FF9E6C;
+    --color-secondary: #FFD8A9;
+}
+
 footer {
     visibility: hidden;
 }
+
+.gradio-container {
+    background: linear-gradient(to right, #FFF4E0, #FFEDDB);
+}
+
+.title {
+    color: #E25822 !important;
+    font-size: 2.5rem !important;
+    font-weight: 700 !important;
+    text-align: center;
+    margin: 1rem 0;
+    text-shadow: 2px 2px 4px rgba(0,0,0,0.1);
+}
+
+.subtitle {
+    color: #2B3A67 !important;
+    font-size: 1.2rem !important;
+    text-align: center;
+    margin-bottom: 2rem;
+}
+
+.model-description {
+    background-color: rgba(255, 255, 255, 0.7);
+    border-radius: 10px;
+    padding: 20px;
+    margin: 20px 0;
+    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+    border-left: 5px solid #E25822;
+}
+
+button.primary {
+    background-color: #E25822 !important;
+}
+
+button:hover {
+    transform: translateY(-2px);
+    box-shadow: 0 5px 15px rgba(0,0,0,0.1);
+}
+
+.tabs {
+    margin-top: 20px;
+}
+
+.gallery {
+    background-color: rgba(255, 255, 255, 0.5);
+    border-radius: 10px;
+    padding: 10px;
+}
 """
 
-with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, analytics_enabled=False) as demo:
-    gr.HTML('<div class="title"> Claude Monet STUDIO </div>')
-    gr.HTML('<div class="title">😄Image to Video Explore: <a href="https://huggingface.co/spaces/ginigen/theater" target="_blank">https://huggingface.co/spaces/ginigen/theater</a></div>')
+with gr.Blocks(theme=gr.themes.Default(), css=custom_css, analytics_enabled=False) as demo:
+    gr.HTML('<div class="title">Pierre-Auguste Renoir STUDIO</div>')
+    gr.HTML('<div class="subtitle">🎨 Image to Video Explore: <a href="https://huggingface.co/spaces/ginigen/theater" target="_blank">https://huggingface.co/spaces/ginigen/theater</a></div>')
+    
+    # Model description with the requested content
+    with gr.Box(elem_classes="model-description"):
+        gr.Markdown("""
+        # About This Model
+        
+        This studio features the **Pierre-Auguste Renoir** artistic style model from [openfree/pierre-auguste-renoir](https://huggingface.co/openfree/pierre-auguste-renoir). 
+        
+        Pierre-Auguste Renoir (1841-1919) was a leading painter in the development of the Impressionist style. His paintings are notable for their vibrant light and saturated color, focusing on people in intimate and candid compositions. The warmth of his palette often emphasized the sensual beauty of his subjects, particularly women.
+        
+        I developed a flux-based learning model trained on a curated collection of high-resolution masterpieces from renowned global artists. This LoRA fine-tuning process leveraged the exceptional quality of open-access imagery released by prestigious institutions including the Art Institute of Chicago. The resulting model demonstrates remarkable capability in capturing the nuanced artistic techniques and stylistic elements across diverse historical art movements.
+        
+        **How to use**: Simply enter a prompt describing a scene in Renoir's style and add [trigger] at the end.
+        """)
 
-    with gr.Tabs() as tabs:
+    with gr.Tabs(elem_classes="tabs") as tabs:
         with gr.Tab("Generation"):
             with gr.Column(elem_id="col-container"):
                 with gr.Row():
@@ -123,10 +191,10 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, analytics_enabled=Fa
                         label="Prompt",
                         show_label=False,
                         max_lines=1,
-                        placeholder="Enter your prompt",
+                        placeholder="Enter your prompt (add [trigger] at the end)",
                         container=False,
                     )
-                    run_button = gr.Button("Run", scale=0)
+                    run_button = gr.Button("Generate", variant="primary", scale=0)
 
                 result = gr.Image(label="Result", show_label=False)
 
@@ -186,25 +254,27 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css, analytics_enabled=Fa
                 )
 
         with gr.Tab("Gallery"):
-            gallery_header = gr.Markdown("### Generated Images Gallery")
+            gallery_header = gr.Markdown("### Your Generated Images")
             generated_gallery = gr.Gallery(
                 label="Generated Images",
-                columns=6,
+                columns=3,
                 show_label=False,
                 value=load_generated_images(),
                 elem_id="generated_gallery",
+                elem_classes="gallery",
                 height="auto"
             )
-            refresh_btn = gr.Button("🔄 Refresh Gallery")
+            refresh_btn = gr.Button("🔄 Refresh Gallery", variant="primary")
 
     # Add sample gallery section at the bottom
-    gr.Markdown("### Claude Monet Style Examples")
+    gr.Markdown("### Pierre-Auguste Renoir Style Examples")
     predefined_gallery = gr.Gallery(
         label="Sample Images", 
         columns=3,
         rows=2,
         show_label=False, 
-        value=load_predefined_images()
+        value=load_predefined_images(),
+        elem_classes="gallery"
     )
 
     # Event handlers