Spaces:
Sleeping
Sleeping
typos in painter types
Browse files- app.py +2 -2
- climategan_wrapper.py +1 -1
app.py
CHANGED
@@ -30,9 +30,9 @@ def predict(cg: ClimateGAN, api_key):
|
|
30 |
img_np = image
|
31 |
|
32 |
painters = {
|
33 |
-
"ClimateGAN Painter": "
|
34 |
"Stable Diffusion Painter": "stable_diffusion",
|
35 |
-
"Both": "
|
36 |
}
|
37 |
|
38 |
output_dict = cg.infer_single(img_np, painters[painter])
|
|
|
30 |
img_np = image
|
31 |
|
32 |
painters = {
|
33 |
+
"ClimateGAN Painter": "climategan",
|
34 |
"Stable Diffusion Painter": "stable_diffusion",
|
35 |
+
"Both": "both",
|
36 |
}
|
37 |
|
38 |
output_dict = cg.infer_single(img_np, painters[painter])
|
climategan_wrapper.py
CHANGED
@@ -304,7 +304,7 @@ class ClimateGAN:
|
|
304 |
], f"Unknown painter: {painter}"
|
305 |
|
306 |
ignore_event = set()
|
307 |
-
if painter == "
|
308 |
ignore_event.add("flood")
|
309 |
|
310 |
# Retrieve numpy events as a dict {event: array[BxHxWxC]}
|
|
|
304 |
], f"Unknown painter: {painter}"
|
305 |
|
306 |
ignore_event = set()
|
307 |
+
if painter == "stable_diffusion":
|
308 |
ignore_event.add("flood")
|
309 |
|
310 |
# Retrieve numpy events as a dict {event: array[BxHxWxC]}
|