AyoAgbaje commited on
Commit
4c543d1
·
verified ·
1 Parent(s): 5eab21d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ import cv2
10
 
11
  def image_predict(img_):
12
  model = load_model('efficientnet_b0.keras')
13
- img = cv2.cvtColor(img_, cv2.COLOR_BGR2RGB)
14
  img = cv2.resize(img, dsize = [224, 224])
15
  img = img / 255.0
16
  img = np.expand_dims(img, axis = 0)
 
10
 
11
  def image_predict(img_):
12
  model = load_model('efficientnet_b0.keras')
13
+ # img = cv2.cvtColor(img_, cv2.COLOR_BGR2RGB)
14
  img = cv2.resize(img, dsize = [224, 224])
15
  img = img / 255.0
16
  img = np.expand_dims(img, axis = 0)