Update app.py

app.py

@@ -8,7 +8,7 @@ model = load_model('Pikachu_and_Raichu.h5')
 labels = ['Pikachu', 'Raichu']
 
 def predict(img):
-    img=img.reshape(-1,224,224,3)
+    img=img.reshape(-1,160,160,3)
     """images_list = []
     images_list.append(np.array(img))
     x = np.asarray(images_list)"""
@@ -17,7 +17,7 @@ def predict(img):
     confidences = {labels[i]: float(prediction[i]) for i in range(2)}
     return confidences
 
-image = gr.inputs.Image(shape=(224, 224))
+image = gr.inputs.Image(shape=(160, 160))
 label = gr.outputs.Label(num_top_classes=2)
 
 gr.Interface(fn=predict, inputs=image,  title="Garbage Classifier",