Update app.py

app.py

@@ -7,21 +7,21 @@ import tensorflow.keras as keras
 from tensorflow.keras.models import load_model
  
 # load model
-model = load_model('mmodel3.h5')
+model = load_model('mmodelinusannekosan.h5')
 
-classnames = ['Pikachu','Raichu','Pichu','Gyaradosu']
+classnames = ['Inu','Neko']
 
 
 
 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(4)}
+  return {classnames[i]: float(prediction[i]) for i in range(2)}
 
                 
 
 image = gr.inputs.Image(shape=(224, 224))
-label = gr.outputs.Label(num_top_classes=4)
+label = gr.outputs.Label(num_top_classes=2)
 
 
 gr.Interface(fn=predict_image, inputs=image,  title="Garbage Classifier V3",