uint8 empty images

app.py

@@ -154,19 +154,19 @@ def predict(cg: ClimateGAN, api_key):
 
         climategan_flood = output_dict.get(
             "climategan_flood",
-            np.ones(input_image.shape) * 255,
+            np.ones(input_image.shape, dtype=np.uint8) * 255,
         )
         stable_flood = output_dict.get(
             "stable_flood",
-            np.ones(input_image.shape) * 255,
+            np.ones(input_image.shape, dtype=np.uint8) * 255,
         )
         stable_copy_flood = output_dict.get(
             "stable_copy_flood",
-            np.ones(input_image.shape) * 255,
+            np.ones(input_image.shape, dtype=np.uint8) * 255,
         )
         concat = output_dict.get(
             "concat",
-            np.ones(input_image.shape) * 255,
+            np.ones(input_image.shape, dtype=np.uint8) * 255,
         )
 
         return (