hash changed

app.py

@@ -4,6 +4,7 @@ import numpy as np
 import glob
 import torch
 import random
+import hashlib
 import imagehash
 from tempfile import NamedTemporaryFile
 from PIL import Image
@@ -86,7 +87,7 @@ tmp_path = '/tmp'
 def arbitrary_to_apose(image, seed):
     # convert image to PIL.Image
     image = Image.fromarray(image)
-    image_hash = str(imagehash.average_hash(image)) + '_' + str(seed)
+    image_hash = str(hashlib.md5(image.tobytes()).hexdigest()) + '_' + str(seed)
     if image_hash not in cache_arbitrary:
         apose_img = infer_api.genStage1(image, seed)
         apose_img.save(f'{tmp_path}/{image_hash}.png')
@@ -101,7 +102,7 @@ def arbitrary_to_apose(image, seed):
 def apose_to_multiview(apose_img, seed):
     # convert image to PIL.Image
     apose_img = Image.fromarray(apose_img)
-    image_hash = str(imagehash.average_hash(apose_img)) + '_' + str(seed)
+    image_hash = str(hashlib.md5(apose_img.tobytes()).hexdigest()) + '_' + str(seed)
     if image_hash not in cache_multiview[0]:
         results = infer_api.genStage2(apose_img, seed, num_levels=1)
         for idx, img in enumerate(results[0]["images"]):