Update app.py

app.py

@@ -1,8 +1,8 @@
+import spaces
 import gradio as gr
 import torch
-from diffusers import StableDiffusionXLPipeline, AutoencoderKL
+from diffusers import FluxPipeline
 from huggingface_hub import hf_hub_download
-import spaces
 from PIL import Image
 import requests
 from translatepy import Translator
@@ -10,8 +10,7 @@ from translatepy import Translator
 translator = Translator()
 
 # Constants
-model = "Corcelio/openvision"
-vae_model = "madebyollin/sdxl-vae-fp16-fix"
+model = "Shakker-Labs/AWPortrait-FL"
 
 CSS = """
 .gradio-container {
@@ -29,15 +28,11 @@ JS = """function () {
   }
 }"""
 
-# Load VAE component
-vae = AutoencoderKL.from_pretrained(
-    vae_model, 
-    torch_dtype=torch.float16
-)
 
 # Ensure model and scheduler are initialized in GPU-enabled function
 if torch.cuda.is_available():
-    pipe = StableDiffusionXLPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16).to("cuda")
+    pipe = FluxPipeline.from_pretrained(model, torch_dtype=torch.bfloat16)
+    pipe.to("cuda")
 
 
 
@@ -46,12 +41,13 @@ if torch.cuda.is_available():
 def generate_image(
     prompt,
     negative="low quality",
-    width=1024,
+    width=768,
     height=1024,
-    scale=1.5,
-    steps=30):
+    scale=3.5,
+    steps=24):
     
     prompt = str(translator.translate(prompt, 'English'))
+    negative_prompt = str(translator.translate(negative, 'English'))
 
     print(f'prompt:{prompt}')
     
@@ -68,26 +64,23 @@ def generate_image(
 
 
 examples = [
-    "a cat eating a piece of cheese",
-    "a ROBOT riding a BLUE horse on Mars, photorealistic",
-    "Ironman VS Hulk, ultrarealistic",
-    "a CUTE robot artist painting on an easel",
-    "Astronaut in a jungle, cold color palette, oil pastel, detailed, 8k",
-    "An alien holding sign board contain word 'Flash', futuristic, neonpunk",
-    "Kids going to school, Anime style"
+    "close up portrait, Amidst the interplay of light and shadows in a photography studio,a soft spotlight traces the contours of a face,highlighting a figure clad in a sleek black turtleneck. The garment,hugging the skin with subtle luxury,complements the Caucasian model's understated makeup,embodying minimalist elegance. Behind,a pale gray backdrop extends,its fine texture shimmering subtly in the dim light,artfully balancing the composition and focusing attention on the subject. In a palette of black,gray,and skin tones,simplicity intertwines with profundity,as every detail whispers untold stories.",
+    "upper body portrait of 1girl wear a black color turtleneck sweater,A proud and confident expression,long hair,look at viewers,studio fashion portrait,studio light,pure white background",
+    "upper body portrait of 1girl wear (red color turtleneck sweater:1),A proud and confident smile expression,long hair,look at viewers,studio fashion portrait,studio light,pure white background",
+    "upper body portrait of 1girl wear suit with tie,A proud and confident smile expression,long hair,look at viewers,studio fashion portrait,studio light,pure white background"
 ]
 
 
 # Gradio Interface
 
 with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
-    gr.HTML("<h1><center>OpenVision</center></h1>")
-    gr.HTML("<p><center><a href='https://huggingface.co/Corcelio/openvision'>OpenVision</a> text-to-image generation</center><br><center>Multi-Languages. Midjourney Aesthetic for All Your Images</center></p>")
+    gr.HTML("<h1><center>Flux</center></h1>")
+    gr.HTML("<p><center><a href='https://huggingface.co/Shakker-Labs/AWPortrait-FL'>Shakker-Labs/AWPortrait-FL</a></center></p>")
     with gr.Group():
         with gr.Row():
-            prompt = gr.Textbox(label='Enter Your Prompt', scale=6)
+            prompt = gr.Textbox(label='Enter Your Prompt(multilingual)', scale=6)
             submit = gr.Button(scale=1, variant='primary')
-    img = gr.Image(label='OpenVision Generated Image')
+    img = gr.Image(label='Flux Generated Image')
     with gr.Accordion("Advanced Options", open=False):
         with gr.Row():
             negative = gr.Textbox(label="Negative prompt", value="low quality")
@@ -97,7 +90,7 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
                 minimum=512,
                 maximum=1280,
                 step=8,
-                value=1024,
+                value=768,
             )
             height = gr.Slider(
                 label="Height",
@@ -112,14 +105,14 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
                 minimum=0,
                 maximum=50,
                 step=0.1,
-                value=1.5,
+                value=3.5,
             )
             steps = gr.Slider(
                 label="Steps",
                 minimum=1,
                 maximum=50,
                 step=1,
-                value=30,
+                value=24,
             )            
     gr.Examples(
         examples=examples,