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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -103,8 +103,8 @@ class WaifuDiffusionInterrogator:
103
  imgByteArr = io.BytesIO()
104
  image.save(imgByteArr, format="png")
105
  imgByteArr = imgByteArr.getvalue()
106
- if imgByteArr in cache:
107
- return cache[imgByteArr]
108
 
109
  full_tags = self._calculation(image)
110
 
@@ -114,8 +114,8 @@ class WaifuDiffusionInterrogator:
114
  # rest are regular tags
115
  tags = dict(full_tags[full_tags['category'] != 9][['name', 'confidence']].values)
116
  self.cache[imgByteArr] = (ratings, tags)
117
- if len(cache) > 25:
118
- cache.popitem(last=False)
119
 
120
  return ratings, tags
121
 
 
103
  imgByteArr = io.BytesIO()
104
  image.save(imgByteArr, format="png")
105
  imgByteArr = imgByteArr.getvalue()
106
+ if imgByteArr in self.cache:
107
+ return self.cache[imgByteArr]
108
 
109
  full_tags = self._calculation(image)
110
 
 
114
  # rest are regular tags
115
  tags = dict(full_tags[full_tags['category'] != 9][['name', 'confidence']].values)
116
  self.cache[imgByteArr] = (ratings, tags)
117
+ if len(self.cache) > 25:
118
+ self.cache.popitem(last=False)
119
 
120
  return ratings, tags
121