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"]
|