Update model.py

model.py

@@ -437,4 +437,4 @@ class Model:
         x_samples_ddim = 255.*x_samples_ddim
         x_samples_ddim = x_samples_ddim.astype(np.uint8)
     
-        return [im_edge, x_samples_ddim]
+        return [im_mask, x_samples_ddim]