Update model.py

model.py

@@ -181,7 +181,7 @@ class Model:
         samples = [
             Image.fromarray(
                 sample.permute(1, 2, 0).cpu().detach().numpy().clip(
-                    0, 255).astype(np.uint8).resize((origin_W, origin_H))) for sample in samples
+                    0, 255).astype(np.uint8)).resize((origin_W, origin_H)) for sample in samples
         ]
         del condition_img
         torch.cuda.empty_cache()
@@ -276,7 +276,7 @@ class Model:
         samples = 255 * (samples * 0.5 + 0.5)
         samples = [
             Image.fromarray(
-                sample.permute(1, 2, 0).cpu().detach().numpy().clip(0, 255).astype(np.uint8).resize((origin_W, origin_H)))
+                sample.permute(1, 2, 0).cpu().detach().numpy().clip(0, 255).astype(np.uint8)).resize((origin_W, origin_H))
             for sample in samples
         ]
         del condition_img