Update app.py

app.py

@@ -24,7 +24,7 @@ del vae
 
 pipe.force_zeros_for_empty_prompt = False
 
-print("Optimizing BRIA 2.2 FAST - this could take a while")
+print("Optimizing BRIA 2.2 HR - this could take a while")
 t=time.time()
 pipe.unet = torch.compile(
     pipe.unet, mode="reduce-overhead", fullgraph=True # 600 secs compilation
@@ -32,7 +32,9 @@ pipe.unet = torch.compile(
 with torch.no_grad():
     outputs = pipe(
         prompt="an apple",
-        num_inference_steps=8,
+        num_inference_steps=30,
+        width=1536,
+        height=1536
     )
 
     # This will avoid future compilations on different shapes
@@ -66,7 +68,7 @@ def infer(prompt,seed,resolution):
 
     w,h = resolution.split()
     w,h = int(w),int(h)
-    image = pipe(prompt,num_inference_steps=8,generator=generator,width=w,height=h).images[0]
+    image = pipe(prompt,num_inference_steps=30,generator=generator,width=w,height=h).images[0]
     print(f'gen time is {time.time()-t} secs')
     
     # Future
@@ -84,12 +86,12 @@ css = """
 """
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
-        gr.Markdown("## BRIA 2.2 FAST")
+        gr.Markdown("## BRIA 2.2 HR")
         gr.HTML('''
           <p style="margin-bottom: 10px; font-size: 94%">
             This is a demo for 
-            <a href="https://huggingface.co/briaai/BRIA-2.2-FAST" target="_blank">BRIA 2.2 FAST </a>. 
-            This is a fast version of BRIA 2.2 text-to-image model, still trained on licensed data, and so provide full legal liability coverage for copyright and privacy infringement.
+            <a href="https://huggingface.co/briaai/BRIA-2.2-HR" target="_blank">BRIA 2.2 HR </a>. 
+            This is a high resolution version of BRIA 2.2 text-to-image model, still trained on licensed data, and so provide full legal liability coverage for copyright and privacy infringement.
           </p>
         ''')
         with gr.Group():
@@ -98,7 +100,7 @@ with gr.Blocks(css=css) as demo:
                 resolution = gr.Dropdown(value=resolutions[0], show_label=True, label="Resolution", choices=resolutions)
                 seed = gr.Textbox(label="Seed", value=-1)
                 submit_btn = gr.Button("Generate")
-        result = gr.Image(label="BRIA 2.2 FAST Result")
+        result = gr.Image(label="BRIA 2.2 HR Result")
 
         # gr.Examples(
         #     examples = [