fix painter selection

app.py

@@ -28,7 +28,14 @@ def predict(cg: ClimateGAN, api_key):
             img_np = io.imread(static_map_url)
         else:
             img_np = image
-        output_dict = cg.infer_single(img_np, painter)
+
+        painters = {
+            "ClimateGAN Painter": "both",
+            "Stable Diffusion Painter": "stable_diffusion",
+            "Both": "climategan",
+        }
+
+        output_dict = cg.infer_single(img_np, painters[painter])
 
         input_image = output_dict["input"]
         masked_input = output_dict["masked_input"]