Update app.py

app.py

@@ -3,29 +3,56 @@ import numpy as np
 import random
 import spaces
 import torch
-from diffusers import  DiffusionPipeline, FlowMatchEulerDiscreteScheduler, AutoencoderTiny, AutoencoderKL
-from transformers import CLIPTextModel, CLIPTokenizer,T5EncoderModel, T5TokenizerFast
+
+from diffusers import DiffusionPipeline, AutoencoderTiny, AutoencoderKL
+from transformers import CLIPTextModel, CLIPTokenizer, T5EncoderModel, T5TokenizerFast
+
 from live_preview_helpers import calculate_shift, retrieve_timesteps, flux_pipe_call_that_returns_an_iterable_of_images
 
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
+# Загружаем автоэнкодер и VAE
 taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype).to(device)
-good_vae = AutoencoderKL.from_pretrained("ifmain/UltraReal_Fine-Tune", subfolder="vae", torch_dtype=dtype).to(device)
-pipe = DiffusionPipeline.from_pretrained("ifmain/UltraReal_Fine-Tune", torch_dtype=dtype, vae=taef1).to(device)
+good_vae = AutoencoderKL.from_pretrained(
+    "ifmain/UltraReal_Fine-Tune",
+    subfolder="vae",
+    torch_dtype=dtype
+).to(device)
+
+# Загружаем основной пайплайн
+pipe = DiffusionPipeline.from_pretrained(
+    "ifmain/UltraReal_Fine-Tune",
+    torch_dtype=dtype,
+    vae=taef1
+).to(device)
+
 torch.cuda.empty_cache()
 
-MAX_SEED = np.iinfo(np.int32).max
-MAX_IMAGE_SIZE = 2048
+# Подключаем LoRA
+pipe.load_lora_weights("ifMain/realism")
 
 pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe)
 
+MAX_SEED = np.iinfo(np.int32).max
+MAX_IMAGE_SIZE = 2048
+
 @spaces.GPU(duration=75)
-def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
+def infer(
+    prompt,
+    seed=42,
+    randomize_seed=False,
+    width=1024,
+    height=1024,
+    guidance_scale=3.5,
+    num_inference_steps=28,
+    progress=gr.Progress(track_tqdm=True)
+):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
+
     generator = torch.Generator().manual_seed(seed)
-    
+
     for img in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
             prompt=prompt,
             guidance_scale=guidance_scale,
@@ -37,14 +64,46 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
             good_vae=good_vae,
         ):
             yield img, seed
-    
-examples = [
-    "d1g1cam, amateur photo, low-lit, Low-resolution photo, shot on a mobile phone, nighttime, noticeable noise in dark areas. young woman, late 20s casually dressed in an oversized pink T-shirt, outdoors, her gaze directed to the side, sad expression, hold cup of coffee, her blonde hair loosely tied back. Her face partially shadowed, minimal makeup visible. she is standing on the street near soviet-era storey buildings, trees, parked cars near. Dull, overcast lighting",
-    "v8s, Dimly lit photo, grungy aesthetic, gritty urban, los angeles city on background, interior of muscle car driving under high speed, amateur quality, nighttime, in motion, smeared background, headlights glowing, noise and grain film, glowing dashboard where speedometer is blurred, rearview mirror, woman's hand with black nailpolish on stearing wheel, first person perspective, pov, wide angle go pro lens",
-    "The image depicts a woman standing outdoors on a snowy path. The individual is wearing a black coat over an orange dress adorned with white floral patterns. They have accessorized with a long, striped scarf in shades of brown and orange that drapes around their neck and shoulders. The woman's feet are bare, suggesting it might be cold outside. In the background, there are multi-story residential buildings with balconies, indicating an urban setting. The ground is partially covered with snow, and the sky appears overcast. photorealistic 8K, film grain, textured skin, dramatic lighting, RAW photo, highly detailed",
+
+# Полные примеры с различными стилями и условиями съемки
+full_examples = [
+    [
+        "d1g1cam, amateur photo, low-lit, Low-resolution photo, shot on a mobile phone, nighttime, noticeable noise in dark areas. "
+        "Young woman, late 20s, casually dressed in an oversized pink T-shirt, outdoors, her gaze directed to the side, sad expression, holding a cup of coffee. "
+        "Her blonde hair is loosely tied back. Her face partially shadowed, minimal makeup visible. She is standing on the street near Soviet-era storey buildings, trees, parked cars nearby. "
+        "Dull, overcast lighting.",
+        1024, 1024, 3.5, 28
+    ],
+    [
+        "v8s, Dimly lit photo, grungy aesthetic, gritty urban, Los Angeles city on background, interior of muscle car driving at high speed. "
+        "Amateur quality, nighttime, in motion, smeared background, headlights glowing, noise and grain film. "
+        "Glowing dashboard where speedometer is blurred, rearview mirror, woman's hand with black nail polish on steering wheel. "
+        "First-person perspective, wide-angle GoPro lens.",
+        1024, 768, 4.0, 30
+    ],
+    [
+        "The image depicts a woman standing outdoors on a snowy path. The individual is wearing a black coat over an orange dress adorned with white floral patterns. "
+        "They have accessorized with a long, striped scarf in shades of brown and orange that drapes around their neck and shoulders. "
+        "The woman's feet are bare, suggesting it might be cold outside. In the background, there are multi-story residential buildings with balconies, indicating an urban setting. "
+        "The ground is partially covered with snow, and the sky appears overcast. "
+        "Photorealistic 8K, film grain, textured skin, dramatic lighting, RAW photo, highly detailed.",
+        1024, 1024, 3.0, 25
+    ],
+    [
+        "Photorealistic close-up portrait of an elderly man with deep wrinkles and a gray beard. His eyes reflect wisdom and experience. "
+        "Soft lighting highlights the texture of his skin. He wears a simple wool sweater. The background is blurred, resembling an indoor wooden interior with bookshelves. "
+        "Film-like quality, 85mm lens, f/1.8 depth of field, cinematic lighting.",
+        768, 1024, 3.8, 35
+    ],
+    [
+        "Golden hour landscape of a countryside road. The sun is setting, casting warm golden light over rolling hills and a winding dirt path. "
+        "Lush green trees line the road, and distant mountains are faintly visible. The sky is a blend of soft pink and deep orange hues. "
+        "Captured on a full-frame DSLR, 50mm f/2.0 lens, soft focus, detailed textures.",
+        1536, 1024, 2.5, 20
+    ],
 ]
 
-css="""
+css = """
 #col-container {
     margin: 0 auto;
     max-width: 520px;
@@ -52,24 +111,23 @@ css="""
 """
 
 with gr.Blocks(css=css) as demo:
-    
     with gr.Column(elem_id="col-container"):
-        gr.Markdown(f"""# UltraReal Fine-Tune (Flux.1 Dev)
-    
-        **🚀 Фотореализм нового уровня!**  
-        Вышла 4-я версия **UltraReal Fine-Tune**, основанная на **Flux.1 Dev**.   
-        Скачать можно тут: [Civitai](https://civitai.com/models/978314?modelVersionId=1413133)  
-    
-        **🚀 Next-level photorealism!**  
-        The 4th version of **UltraReal Fine-Tune**, based on **Flux.1 Dev**, has been released.
-        You can download it here: [Civitai](https://civitai.com/models/978314?modelVersionId=1413133)  
-        
-        [[model](https://huggingface.co/black-forest-labs/FLUX.1-dev)] [[non-commercial license](https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md)] [[blog](https://blackforestlabs.ai/announcing-black-forest-labs/)]
-        """)
-
-        
+        gr.Markdown(
+            """# UltraReal Fine-Tune (Flux.1 Dev)  
+
+            **🚀 Фотореализм нового уровня!**  
+            Вышла 4-я версия **UltraReal Fine-Tune**, основанная на **Flux.1 Dev**.  
+            Скачать можно тут: [Civitai](https://civitai.com/models/978314?modelVersionId=1413133)  
+
+            **🚀 Next-level photorealism!**  
+            The 4th version of **UltraReal Fine-Tune**, based on **Flux.1 Dev**, has been released.  
+            You can download it here: [Civitai](https://civitai.com/models/978314?modelVersionId=1413133)  
+
+            [[model](https://huggingface.co/black-forest-labs/FLUX.1-dev)] [[non-commercial license](https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md)] [[blog](https://blackforestlabs.ai/announcing-black-forest-labs/)]
+            """
+        )
+
         with gr.Row():
-            
             prompt = gr.Text(
                 label="Prompt",
                 show_label=False,
@@ -77,13 +135,11 @@ with gr.Blocks(css=css) as demo:
                 placeholder="Enter your prompt",
                 container=False,
             )
-            
             run_button = gr.Button("Run", scale=0)
-        
+
         result = gr.Image(label="Result", show_label=False)
-        
+
         with gr.Accordion("Advanced Settings", open=False):
-            
             seed = gr.Slider(
                 label="Seed",
                 minimum=0,
@@ -91,11 +147,9 @@ with gr.Blocks(css=css) as demo:
                 step=1,
                 value=0,
             )
-            
             randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
-            
+
             with gr.Row():
-                
                 width = gr.Slider(
                     label="Width",
                     minimum=256,
@@ -103,7 +157,6 @@ with gr.Blocks(css=css) as demo:
                     step=32,
                     value=1024,
                 )
-                
                 height = gr.Slider(
                     label="Height",
                     minimum=256,
@@ -111,9 +164,8 @@ with gr.Blocks(css=css) as demo:
                     step=32,
                     value=1024,
                 )
-            
-            with gr.Row():
 
+            with gr.Row():
                 guidance_scale = gr.Slider(
                     label="Guidance Scale",
                     minimum=1,
@@ -121,7 +173,6 @@ with gr.Blocks(css=css) as demo:
                     step=0.1,
                     value=3.5,
                 )
-  
                 num_inference_steps = gr.Slider(
                     label="Number of inference steps",
                     minimum=1,
@@ -129,20 +180,19 @@ with gr.Blocks(css=css) as demo:
                     step=1,
                     value=28,
                 )
-        
+
         gr.Examples(
-            examples = examples,
-            fn = infer,
-            inputs = [prompt],
-            outputs = [result, seed],
+            examples=full_examples,
+            inputs=[prompt, width, height, guidance_scale, num_inference_steps],
+            outputs=[result, seed],
             cache_examples="lazy"
         )
 
     gr.on(
         triggers=[run_button.click, prompt.submit],
-        fn = infer,
-        inputs = [prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
-        outputs = [result, seed]
+        fn=infer,
+        inputs=[prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
+        outputs=[result, seed]
     )
 
 demo.launch()