Upload app.py

app.py

@@ -475,9 +475,10 @@ def generate_with_prompt_style_guidance(prompt, style, seed,num_inference_steps,
                 loss_scale = 200 #
                 loss = ymca_loss(denoised_images) * loss_scale
             elif loss_function == "cmyk":
-                loss_scale = 10 #
+                loss_scale = 1 #
                 loss = cymk_loss(denoised_images) * loss_scale
             else :
+                loss_scale = 200
                 loss = ymca_loss(denoised_images) * loss_scale
 
             # # Occasionally print it out
@@ -542,7 +543,7 @@ demo = gr.Interface(inference,
                                 step=8,
                                 label="Select Guidance Scale",
                                 interactive=True,
-                                ),gr.Radio(["contrast", "blue_original", "blue_modified","ymca","cmyk"], label="loss-function", info="loss-function" , value="ymca_loss"),
+                                ),gr.Radio(["contrast", "blue_original", "blue_modified","ymca","cmyk"], label="loss-function", info="loss-function" , value="ymca"),
                               ],
                     outputs = [
                               gr.Image(label="Stable Diffusion Output"),