Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import torch
-from diffusers import StableDiffusionXLPipeline, AutoencoderKL
+from diffusers import StableDiffusionXLPipeline, AutoencoderKL, KDPM2AncestralDiscreteScheduler
 from huggingface_hub import hf_hub_download
 import spaces
 from PIL import Image
@@ -39,6 +39,7 @@ vae = AutoencoderKL.from_pretrained(
 if torch.cuda.is_available():
     pipe = StableDiffusionXLPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16).to("cuda")
 
+pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 
 
 # Function 
@@ -49,7 +50,8 @@ def generate_image(
     width=1024,
     height=1024,
     scale=1.5,
-    steps=30):
+    steps=30,
+    clip=3):
     
     prompt = str(translator.translate(prompt, 'English'))
 
@@ -62,8 +64,8 @@ def generate_image(
         height=height,
         guidance_scale=scale,
         num_inference_steps=steps,
+        clip_skip=clip,
     )
-    print(image.images[0])
     return image.images[0]
 
 
@@ -85,7 +87,7 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
     gr.HTML("<p><center><a href='https://huggingface.co/Corcelio/openvision'>OpenVision</a> text-to-image generation</center><br><center>Multi-Languages. Midjourney Aesthetic for All Your Images</center></p>")
     with gr.Group():
         with gr.Row():
-            prompt = gr.Textbox(label='Enter Your Prompt', scale=6)
+            prompt = gr.Textbox(label='Enter Your Prompt', value="best quality, HD, aesthetic", scale=6)
             submit = gr.Button(scale=1, variant='primary')
     img = gr.Image(label='OpenVision Generated Image')
     with gr.Accordion("Advanced Options", open=False):
@@ -108,19 +110,26 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
             )
         with gr.Row():
             scale = gr.Slider(
-                label="Guidance Scale",
-                minimum=0,
-                maximum=50,
+                label="Guidance",
+                minimum=3.5,
+                maximum=7,
                 step=0.1,
-                value=1.5,
+                value=7,
             )
             steps = gr.Slider(
                 label="Steps",
                 minimum=1,
                 maximum=50,
                 step=1,
-                value=30,
-            )            
+                value=50,
+            )
+            clip = gr.Slider(
+                label="Clip Skip",
+                minimum=1,
+                maximum=10,
+                step=1,
+                value=3,
+            )      
     gr.Examples(
         examples=examples,
         inputs=prompt,
@@ -130,11 +139,11 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
     )
 
     prompt.submit(fn=generate_image,
-                 inputs=[prompt, negative, width, height, scale, steps],
+                 inputs=[prompt, negative, width, height, scale, steps, clip],
                  outputs=img,
                  )
     submit.click(fn=generate_image,
-                 inputs=[prompt, negative, width, height, scale, steps],
+                 inputs=[prompt, negative, width, height, scale, steps, clip],
                  outputs=img,
                  )