Update app.py

app.py

@@ -6,7 +6,26 @@ import diffusers
 from share_btn import community_icon_html, loading_icon_html, share_js
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = AutoPipelineForInpainting.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", torch_dtype=torch.float16, variant="fp16").to(device)
+unet = UNet2DConditionModel.from_pretrained(
+    "briaai/BRIA-2.3-Inpainting",
+    subfolder="unet",
+    torch_dtype=torch.float16,
+)
+
+scheduler = DDIMScheduler.from_pretrained("briaai/BRIA-2.3", subfolder="scheduler",clip_sample=False)
+
+pipe = StableDiffusionXLInpaintPipeline.from_pretrained(
+    "briaai/BRIA-2.3",
+    unet=unet,
+    scheduler=scheduler,
+    torch_dtype=torch.float16,
+    force_zeros_for_empty_prompt=False
+)
+pipe = pipe.to(device)
+pipe.force_zeros_for_empty_prompt = False
+
+default_negative_prompt= "Logo,Watermark,Text,Ugly,Morbid,Extra fingers,Poorly drawn hands,Mutation,Blurry,Extra limbs,Gross proportions,Missing arms,Mutated hands,Long neck,Duplicate,Mutilated,Mutilated hands,Poorly drawn face,Deformed,Bad anatomy,Cloned face,Malformed limbs,Missing legs,Too many fingers"
+
 
 def read_content(file_path: str) -> str:
     """read the content of target file
@@ -16,20 +35,11 @@ def read_content(file_path: str) -> str:
 
     return content
 
+@spaces.GPU()
 def predict(dict, prompt="", negative_prompt="", guidance_scale=7.5, steps=20, strength=1.0, scheduler="EulerDiscreteScheduler"):
     if negative_prompt == "":
         negative_prompt = None
-    scheduler_class_name = scheduler.split("-")[0]
 
-    add_kwargs = {}
-    if len(scheduler.split("-")) > 1:
-        add_kwargs["use_karras"] = True
-    if len(scheduler.split("-")) > 2:
-        add_kwargs["algorithm_type"] = "sde-dpmsolver++"
-
-    scheduler = getattr(diffusers, scheduler_class_name)
-    pipe.scheduler = scheduler.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", subfolder="scheduler", **add_kwargs)
-    
     init_image = dict["image"].convert("RGB").resize((1024, 1024))
     mask = dict["mask"].convert("RGB").resize((1024, 1024))
     
@@ -77,7 +87,15 @@ div#share-btn-container > div {flex-direction: row;background: black;align-items
 
 image_blocks = gr.Blocks(css=css, elem_id="total-container")
 with image_blocks as demo:
-    gr.HTML(read_content("header.html"))
+    with gr.Column(elem_id="col-container"):
+        gr.Markdown("## BRIA 2.3")
+        gr.HTML('''
+          <p style="margin-bottom: 10px; font-size: 94%">
+            This is a demo for 
+            <a href="https://huggingface.co/briaai/BRIA-2.3" target="_blank">BRIA 2.3 text-to-image </a>. 
+            BRIA 2.3 improve the generation of humans and illustrations compared to BRIA 2.2 while still trained on licensed data, and so provide full legal liability coverage for copyright and privacy infringement.
+          </p>
+        ''')
     with gr.Row():
                 with gr.Column():
                     image = gr.Image(source='upload', tool='sketch', elem_id="image_upload", type="pil", label="Upload",height=400)
@@ -88,13 +106,11 @@ with image_blocks as demo:
                     
                     with gr.Accordion(label="Advanced Settings", open=False):
                         with gr.Row(mobile_collapse=False, equal_height=True):
-                            guidance_scale = gr.Number(value=7.5, minimum=1.0, maximum=20.0, step=0.1, label="guidance_scale")
-                            steps = gr.Number(value=20, minimum=10, maximum=30, step=1, label="steps")
+                            guidance_scale = gr.Number(value=7.5, minimum=1.0, maximum=10.0, step=0.5, label="guidance_scale")
+                            steps = gr.Number(value=30, minimum=20, maximum=50, step=1, label="steps")
                             strength = gr.Number(value=0.99, minimum=0.01, maximum=1.0, step=0.01, label="strength")
-                            negative_prompt = gr.Textbox(label="negative_prompt", placeholder="Your negative prompt", info="what you don't want to see in the image")
-                        with gr.Row(mobile_collapse=False, equal_height=True):
-                            schedulers = ["DEISMultistepScheduler", "HeunDiscreteScheduler", "EulerDiscreteScheduler", "DPMSolverMultistepScheduler", "DPMSolverMultistepScheduler-Karras", "DPMSolverMultistepScheduler-Karras-SDE"]
-                            scheduler = gr.Dropdown(label="Schedulers", choices=schedulers, value="EulerDiscreteScheduler")
+                            negative_prompt = gr.Textbox(label="negative_prompt", value=default_negative_prompt, placeholder=default_negative_prompt, info="what you don't want to see in the image")
+
                         
                 with gr.Column():
                     image_out = gr.Image(label="Output", elem_id="output-img", height=400)
@@ -110,16 +126,7 @@ with image_blocks as demo:
 
     gr.Examples(
                 examples=[
-                    ["./imgs/aaa (8).png"],
-                    ["./imgs/download (1).jpeg"],
-                    ["./imgs/0_oE0mLhfhtS_3Nfm2.png"],
-                    ["./imgs/02_HubertyBlog-1-1024x1024.jpg"],
-                    ["./imgs/jdn_jacques_de_nuce-1024x1024.jpg"],
-                    ["./imgs/c4ca473acde04280d44128ad8ee09e8a.jpg"],
-                    ["./imgs/canam-electric-motorcycles-scaled.jpg"],
-                    ["./imgs/e8717ce80b394d1b9a610d04a1decd3a.jpeg"],
-                    ["./imgs/Nature___Mountains_Big_Mountain_018453_31.jpg"],
-                    ["./imgs/Multible-sharing-room_ccexpress-2-1024x1024.jpeg"],
+                    ["./imgs/example.png"],
                 ],
                 fn=predict,
                 inputs=[image],