typos in painter types

app.py

@@ -30,9 +30,9 @@ def predict(cg: ClimateGAN, api_key):
             img_np = image
 
         painters = {
-            "ClimateGAN Painter": "both",
+            "ClimateGAN Painter": "climategan",
             "Stable Diffusion Painter": "stable_diffusion",
-            "Both": "climategan",
+            "Both": "both",
         }
 
         output_dict = cg.infer_single(img_np, painters[painter])

climategan_wrapper.py

@@ -304,7 +304,7 @@ class ClimateGAN:
         ], f"Unknown painter: {painter}"
 
         ignore_event = set()
-        if painter == "climategan":
+        if painter == "stable_diffusion":
             ignore_event.add("flood")
 
         # Retrieve numpy events as a dict {event: array[BxHxWxC]}