Spaces:
Runtime error
Runtime error
ignore no-copy stable diff output for concat
Browse files
app.py
CHANGED
@@ -148,7 +148,17 @@ def predict(cg: ClimateGAN, api_key):
|
|
148 |
"Both": "both",
|
149 |
}
|
150 |
print("Using painter", painters[painter])
|
151 |
-
output_dict = cg.infer_single(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
152 |
|
153 |
input_image = output_dict["input"]
|
154 |
masked_input = output_dict["masked_input"]
|
|
|
148 |
"Both": "both",
|
149 |
}
|
150 |
print("Using painter", painters[painter])
|
151 |
+
output_dict = cg.infer_single(
|
152 |
+
img_np,
|
153 |
+
painters[painter],
|
154 |
+
concats=[
|
155 |
+
"input",
|
156 |
+
"masked_input",
|
157 |
+
"climategan_flood",
|
158 |
+
"stable_copy_flood",
|
159 |
+
],
|
160 |
+
as_pil_image=True,
|
161 |
+
)
|
162 |
|
163 |
input_image = output_dict["input"]
|
164 |
masked_input = output_dict["masked_input"]
|