pixel diff bug resolved

app.py

@@ -13,7 +13,7 @@ from attribution import MappingNetwork
 
 import math
 from typing import List
-from PIL import Image
+from PIL import Image, ImageChops
 import numpy as np
 import torch
 
@@ -79,9 +79,11 @@ class AttributionModel:
         image_org = self.inference_without_attribution(out_latents)
         image_org_pil = self.pipe.numpy_to_pil(image_org[0])
 
-        image_diff_pil = self.pipe.numpy_to_pil(image_attr[0] - image_org[0])
+        # image_diff_pil = self.pipe.numpy_to_pil(image_attr[0] - image_org[0])
+        diff_factor = 5
+        image_diff_pil = ImageChops.difference(image_org_pil[0], image_attr_pil[0]).convert("RGB", (diff_factor,0,0,0,0,diff_factor,0,0,0,0,diff_factor,0))
 
-        return image_org_pil[0], image_attr_pil[0], image_diff_pil[0]
+        return image_org_pil[0], image_attr_pil[0], image_diff_pil
 
     def inference_without_attribution(self, latents):
         latents = 1 / 0.18215 * latents