Minor fix

app.py

@@ -151,29 +151,29 @@ def generate_with_embs(num_inference_steps, guidance_scale, seed, text_input, te
 
     return latents_to_pil(latents)[0]
 
-def guide_loss(images, loss_type='grayscale'):
+def guide_loss(images, loss_type='Gayscale'):
     # grayscale loss
-    if loss_type == 'grayscale':
+    if loss_type == 'Grayscale':
       transformed_imgs = grayscale_transformer(images)
       error = torch.abs(transformed_imgs - images).mean()
 
     # brightness loss
-    elif loss_type == 'bright':
+    elif loss_type == 'Bright':
       transformed_imgs = tfms.functional.adjust_brightness(images, brightness_factor=3)
       error = torch.abs(transformed_imgs - images).mean()
 
     # contrast loss
-    elif loss_type == 'contrast':
+    elif loss_type == 'Contrast':
       transformed_imgs = tfms.functional.adjust_contrast(images, contrast_factor=10)
       error = torch.abs(transformed_imgs - images).mean()
 
     # symmetry loss - Flip the image along the width
-    elif loss_type == "symmetry":
+    elif loss_type == "Symmetry":
       flipped_image = torch.flip(images, [3])
       error = F.mse_loss(images, flipped_image)
 
     # saturation loss
-    elif loss_type == 'saturation':
+    elif loss_type == 'Saturation':
       transformed_imgs = tfms.functional.adjust_saturation(images,saturation_factor = 10)
       error = torch.abs(transformed_imgs - images).mean()
 
@@ -291,7 +291,7 @@ demo = gr.Interface(inference,
                               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="Concept"),
+                                                                  'Symmetry', 'Saturation'], value="Grayscale"),
                               gr.Slider(100, 10000, 100, step = 1, label="Loss scale")],
                     outputs= [gr.Image(width=320, height=320, label="Generated art"),
                               gr.Image(width=320, height=320, label="Generated art with guidance")],