Update app.py

app.py

@@ -6,7 +6,7 @@ import torch
 from PIL import Image
 import random
 
-model_id = 'aipicasso/cool-japan-diffusion-2-1-1-1'
+model_id = 'aipicasso/cool-japan-diffusion-2-1-2-beta'
 
 scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
 feature_extractor = CLIPFeatureExtractor.from_pretrained(model_id)
@@ -41,14 +41,14 @@ def inference(prompt, guidance, steps, image_size="Square", seed=0, img=None, st
   prompt,neg_prompt=auto_prompt_correction(prompt,neg_prompt,cool_japan_type,disable_auto_prompt_correction)
 
   if(image_size=="Portrait"):
-      height=768
-      width=576
+      height=1024
+      width=768
   elif(image_size=="Landscape"):
-      height=576
-      width=768  
+      height=768
+      width=1024  
   else:
-      height=512
-      width=512
+      height=768
+      width=768
 
   print(prompt,neg_prompt)
     
@@ -149,13 +149,13 @@ with gr.Blocks(css=css) as demo:
         f"""
             <div class="main-div">
               <div>
-                <h1>Cool Japan Diffusion 2.1.1.1</h1>
+                <h1>Cool Japan Diffusion 2.1.2 Beta</h1>
               </div>
               <p>
-               Demo for <a href="https://huggingface.co/aipicasso/cool-japan-diffusion-2-1-1">Cool Japan Diffusion 2.1.1.1</a> Stable Diffusion model.<br>
+               Demo for <a href="https://huggingface.co/aipicasso/cool-japan-diffusion-2-1-2-beta">Cool Japan Diffusion 2.1.2 Beta</a> Stable Diffusion model.<br>
               </p>
               <p>
-              sample : Click "Generate" button without any scripts.
+              sample : Click "Generate" button without any prompts.
               </p>
               <p>
               sample prompt1 : girl, kimono
@@ -184,7 +184,7 @@ with gr.Blocks(css=css) as demo:
                 prompt = gr.Textbox(label="Prompt", show_label=False, max_lines=2,placeholder="[your prompt]").style(container=False)
                 generate = gr.Button(value="Generate").style(rounded=(False, True, True, False))
 
-              image_out = gr.Image(height=768,width=576)
+              image_out = gr.Image(height=768,width=768)
           error_output = gr.Markdown()
 
         with gr.Column(scale=45):
@@ -195,7 +195,7 @@ with gr.Blocks(css=css) as demo:
               with gr.Row():
                 image_size=gr.Radio(["Portrait","Landscape","Square"])
                 image_size.show_label=False
-                image_size.value="Portrait"
+                image_size.value="Square"
                 
               with gr.Row():
                 guidance = gr.Slider(label="Guidance scale", value=7.5, maximum=15)