Update app.py

app.py

@@ -21,7 +21,7 @@ def style_transfer(content_img, style_strength, style_img_1 = None, iw_1 = 0, st
       if style_img is not None:
         styles.append(transform(style_img).unsqueeze(0))
     if preserve_color == "None": preserve_color = None
-    elif preserve_color == "Whitening": preserve_color = "batch_wct"
+    elif preserve_color == "Whitening & Coloring": preserve_color = "batch_wct"
     elif preserve_color == "Histogram matching": preserve_color = "histogram_matching"
     with torch.no_grad():
         stylized_img = model(content, styles, style_strength, interpolation_weights, preserve_color = preserve_color)