atticus commited on
Commit
4fa1291
1 Parent(s): 896cee4
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -141,6 +141,9 @@ def search(mode, method, image, text):
141
  if b:
142
  res.append(img)
143
  idx += 1
 
 
 
144
  return res
145
 
146
  if __name__ == "__main__":
141
  if b:
142
  res.append(img)
143
  idx += 1
144
+ while idx < topK:
145
+ res.append(np.zeros((255, 255, 3)))
146
+ idx += 1
147
  return res
148
 
149
  if __name__ == "__main__":