vict0rsch commited on
Commit
0776dea
1 Parent(s): 7d48feb

ignore no-copy stable diff output for concat

Browse files
Files changed (1) hide show
  1. app.py +11 -1
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(img_np, painters[painter], as_pil_image=True)
 
 
 
 
 
 
 
 
 
 
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"]