offry commited on
Commit
713abf1
·
verified ·
1 Parent(s): 60449ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ def create_retrieval_figure(res):
34
  image = cv2.imdecode(np.frombuffer(imgfile, np.uint8), 1)
35
  # image_resized = cv2.resize(image, (224, 224), interpolation=cv2.INTER_LINEAR)
36
  except Exception:
37
- image = cv2.imdecode(np.array((3, 224, 224), np.uint8), 1)
38
  ax = fig.add_subplot(rows, cols, i + 1)
39
  plt.axis('off')
40
  plt.imshow(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
 
34
  image = cv2.imdecode(np.frombuffer(imgfile, np.uint8), 1)
35
  # image_resized = cv2.resize(image, (224, 224), interpolation=cv2.INTER_LINEAR)
36
  except Exception:
37
+ image = cv2.imdecode(np.array((224, 224, 3), np.uint8), 1)
38
  ax = fig.add_subplot(rows, cols, i + 1)
39
  plt.axis('off')
40
  plt.imshow(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))