Updated defaults

app.py

@@ -221,7 +221,7 @@ def generate_with_guide_loss(num_inference_steps, guidance_scale, seed, text_inp
         noise_pred = noise_pred_uncond + guidance_scale * (noise_pred_text - noise_pred_uncond)
 
         #### ADDITIONAL GUIDANCE ###
-        if i%5 == 0:
+        if i%3 == 0:
             # Requires grad on the latents
             latents = latents.detach().requires_grad_()
 
@@ -236,7 +236,7 @@ def generate_with_guide_loss(num_inference_steps, guidance_scale, seed, text_inp
             loss = guide_loss(denoised_images, loss_type) * loss_scale
 
             # Occasionally print it out
-            if i%10==0:
+            if i%5==0:
                 print(i, 'loss:', loss.item())
 
             # Get gradient
@@ -277,22 +277,22 @@ def inference(text, style, inference_step, guidance_scale, seed, guidance_method
 
 title = "Stable Diffusion with Textual Inversion"
 description = "A simple Gradio interface to infer Stable Diffusion and generate images with different art style"
-examples = [["A sweet potato farm", 'Concept', 10, 1.5, 1, 'Grayscale', 100],
-            ["Sky full of cotton candy", 'Realistic', 10, 3.5, 2, 'Bright', 200],
-            ["Kittens in the bathtub", 'Line', 10, 5.5, 3, 'Contrast', 300],
-            ["Water skiing on a lake", 'Ricky', 10, 7.5, 4, 'Symmetry', 400],
-            ["Miniature pet elephant", 'Plane Scape', 10, 9.5, 5, 'Saturation', 500]]
+examples = [["A sweet potato farm", 'Concept', 6, 1.5, 1, 'Grayscale', 100],
+            ["Sky full of cotton candy", 'Realistic', 6, 3.5, 2, 'Bright', 200],
+            ["Kittens in the bathtub", 'Line', 6, 5.5, 3, 'Contrast', 300],
+            ["Water skiing on a lake", 'Ricky', 6, 7.5, 4, 'Symmetry', 400],
+            ["Miniature pet elephant", 'Plane Scape', 6, 9.5, 5, 'Saturation', 500]]
 
 demo = gr.Interface(inference, 
                     inputs = [gr.Textbox(label="Prompt", type="text"),
                               gr.Dropdown(label="Style", choices=['Concept', 'Realistic', 'Line', 
                                                                   'Ricky', 'Plane Scape'], value="Concept"), 
-                              gr.Slider(10, 30, 10, step = 5, label="Inference steps"),
+                              gr.Slider(10, 30, 6, step = 1, label="Inference steps"),
                               gr.Slider(1, 10, 7.5, step = 0.1, label="Guidance scale"),
                               gr.Slider(0, 10000, 1, step = 1, label="Seed"),
                               gr.Dropdown(label="Guidance method", choices=['Grayscale', 'Bright', 'Contrast', 
                                                                   'Symmetry', 'Saturation'], value="Grayscale"),
-                              gr.Slider(100, 10000, 100, step = 1, label="Loss scale")],
+                              gr.Slider(100, 10000, 200, step = 100, label="Loss scale")],
                     outputs= [gr.Image(width=320, height=320, label="Generated art"),
                               gr.Image(width=320, height=320, label="Generated art with guidance")],
                     title=title,