Update app.py

app.py

@@ -11,9 +11,7 @@ resolutions = ["1024 1024","1280 768","1344 768","768 1344","768 1280"]
 # Ng
 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"
 
-# Load pipeline
-# trust_remote_code = True - allows loading a transformer which is not present at the transformers library(from transformer/bria_transformer.py)
-pipe = BriaPipeline.from_pretrained("briaai/BRIA-3.0-TOUCAN", torch_dtype=torch.bfloat16,trust_remote_code=True)
+pipe = BriaPipeline.from_pretrained("briaai/BRIA-3.1", torch_dtype=torch.bfloat16,trust_remote_code=True)
 pipe.to(device="cuda")
 
 @spaces.GPU(enable_queue=True)
@@ -55,12 +53,12 @@ css = """
 """
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
-        gr.Markdown("## BRIA 3.0-TOUCAN")
+        gr.Markdown("## BRIA 3.1")
         gr.HTML('''
           <p style="margin-bottom: 10px; font-size: 94%">
             This is a demo for 
-            <a href="https://huggingface.co/briaai/BRIA-3.0-TOUCAN" target="_blank">BRIA 3.0-TOUCAN text-to-image </a>. 
-            BRIA 3.0-TOUCAN improve the generation of humans and illustrations compared to BRIA 2.3 while still trained on licensed data, and so provide full legal liability coverage for copyright and privacy infringement.
+            <a href="https://huggingface.co/briaai/BRIA-3.1" target="_blank">BRIA 3.1 text-to-image </a>. 
+            BRIA 3.1 improve the generation of humans and illustrations compared to BRIA 2.3 while still trained on licensed data, and so provide full legal liability coverage for copyright and privacy infringement.
           </p>
         ''')
         with gr.Group():
@@ -70,7 +68,7 @@ with gr.Blocks(css=css) as demo:
                 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.0-TOUCAN Result")
+        result = gr.Image(label="BRIA-3.1 Result")
 
         # gr.Examples(
         #     examples = [