Spaces:

yupikopi
/

test9

Runtime error

yupikopi commited on Apr 6, 2023

Commit

fa4978f

1 Parent(s): 952f5b1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ model = load_model('Pikachu_and_Raichu.h5')
 labels = ['Pikachu', 'Raichu']
 def predict(img):
-    img=img.reshape(-1,160,160,3)
     """images_list = []
     images_list.append(np.array(img))
     x = np.asarray(images_list)"""

 labels = ['Pikachu', 'Raichu']
 def predict(img):
+    img=img.reshape(160,160,3)
     """images_list = []
     images_list.append(np.array(img))
     x = np.asarray(images_list)"""