yupikopi commited on
Commit
fa4978f
1 Parent(s): 952f5b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ model = load_model('Pikachu_and_Raichu.h5')
8
  labels = ['Pikachu', 'Raichu']
9
 
10
  def predict(img):
11
- img=img.reshape(-1,160,160,3)
12
  """images_list = []
13
  images_list.append(np.array(img))
14
  x = np.asarray(images_list)"""
 
8
  labels = ['Pikachu', 'Raichu']
9
 
10
  def predict(img):
11
+ img=img.reshape(160,160,3)
12
  """images_list = []
13
  images_list.append(np.array(img))
14
  x = np.asarray(images_list)"""