Update app.py

app.py

@@ -9,19 +9,19 @@ from tensorflow.keras.models import load_model
 # load model
 model = load_model('mmodel.h5')
 
-classnames = ['Pikachu','Raichu','Pichu']
+classnames = ['Pikachu','Raichu','Pichu','Gyaradosu']
 
 
 
 def predict_image(img):
   img_4d=img.reshape(-1,224, 224,3)
   prediction=model.predict(img_4d)[0]
-  return {classnames[i]: float(prediction[i]) for i in range(3)}
+  return {classnames[i]: float(prediction[i]) for i in range(4)}
 
                 
 
 image = gr.inputs.Image(shape=(224, 224))
-label = gr.outputs.Label(num_top_classes=3)
+label = gr.outputs.Label(num_top_classes=4)
 
 
 gr.Interface(fn=predict_image, inputs=image,  title="Garbage Classifier V3",