Update app.py

app.py

@@ -11,7 +11,8 @@ resolutions = ["1024 1024","1280 768","1344 768","768 1344","768 1280"]
 # Ng
 default_negative_prompt= "Logo,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"
 
-pipe = BriaPipeline.from_pretrained("briaai/BRIA-3.1", torch_dtype=torch.bfloat16,trust_remote_code=True)
+transformer = BriaTransformer2DModel.from_pretrained("briaai/BRIA-3.2",subfolder='transformer',torch_dtype=torch.bfloat16)
+pipe = BriaPipeline.from_pretrained("briaai/BRIA-3.1", transformer=transformer, torch_dtype=torch.bfloat16,trust_remote_code=True)
 pipe.to(device="cuda")
 
 @spaces.GPU(enable_queue=True)
@@ -53,11 +54,11 @@ css = """
 """
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
-        gr.Markdown("## BRIA 3.1")
+        gr.Markdown("## BRIA 3.2")
         gr.HTML('''
           <p style="margin-bottom: 10px; font-size: 94%">
             This is a demo for 
-            <a href="https://huggingface.co/briaai/BRIA-3.1" target="_blank">BRIA 3.1 text-to-image </a>. 
+            <a href="https://huggingface.co/briaai/BRIA-3.2" target="_blank">BRIA 3.2 text-to-image </a>. 
 is our new text-to-image model that achieves high-quality generation while being trained exclusively on fully licensed data. We offer both API access and direct access to the model weights, making integration seamless for developers.         </p>
         ''')
         with gr.Group():
@@ -67,7 +68,7 @@ is our new text-to-image model that achieves high-quality generation while being
                 seed = gr.Textbox(label="Seed", value=-1)
                 negative_prompt = gr.Textbox(label="Negative Prompt", value=default_negative_prompt)
                 submit_btn = gr.Button("Generate")
-        result = gr.Image(label="BRIA-3.1 Result")
+        result = gr.Image(label="BRIA-3.2 Result")
 
         # gr.Examples(
         #     examples = [