Spaces:
Running
Running
Update app.py
Browse files
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=
|
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]
|