yoinked commited on
Commit
7d66afc
1 Parent(s): 4b2b363

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -101,7 +101,7 @@ class WaifuDiffusionInterrogator:
101
  def interrogate(self, image: Image) -> Tuple[Dict[str, float], Dict[str, float]]:
102
 
103
  imgByteArr = io.BytesIO()
104
- image.save(imgByteArr, format=image.format)
105
  imgByteArr = imgByteArr.getvalue()
106
  if imgByteArr in cache:
107
  return cache[imgByteArr]
 
101
  def interrogate(self, image: Image) -> Tuple[Dict[str, float], Dict[str, float]]:
102
 
103
  imgByteArr = io.BytesIO()
104
+ image.save(imgByteArr, format="png")
105
  imgByteArr = imgByteArr.getvalue()
106
  if imgByteArr in cache:
107
  return cache[imgByteArr]