Spaces:
Runtime error
Runtime error
add opencv-python
Browse files- climategan_wrapper.py +5 -1
- requirements.txt +1 -0
climategan_wrapper.py
CHANGED
@@ -576,7 +576,11 @@ if __name__ == "__main__":
|
|
576 |
cg = ClimateGAN("models/climategan")
|
577 |
|
578 |
# check painter type
|
579 |
-
assert args.painter in {
|
|
|
|
|
|
|
|
|
580 |
f"Unknown painter {args.painter}. "
|
581 |
+ "Allowed values are 'climategan', 'stable_diffusion' and 'both'."
|
582 |
)
|
|
|
576 |
cg = ClimateGAN("models/climategan")
|
577 |
|
578 |
# check painter type
|
579 |
+
assert args.painter in {
|
580 |
+
"climategan",
|
581 |
+
"stable_diffusion",
|
582 |
+
"both",
|
583 |
+
}, (
|
584 |
f"Unknown painter {args.painter}. "
|
585 |
+ "Allowed values are 'climategan', 'stable_diffusion' and 'both'."
|
586 |
)
|
requirements.txt
CHANGED
@@ -87,6 +87,7 @@ networkx
|
|
87 |
numpy
|
88 |
nvidia-ml-py3
|
89 |
omegaconf
|
|
|
90 |
orjson
|
91 |
packaging
|
92 |
pandas
|
|
|
87 |
numpy
|
88 |
nvidia-ml-py3
|
89 |
omegaconf
|
90 |
+
opencv-python
|
91 |
orjson
|
92 |
packaging
|
93 |
pandas
|