Update app.py

app.py

@@ -16,7 +16,7 @@ classnames = ['neko','inu']
 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(2)}
+  return {classnames[i]: float(prediction[i]) for i in range(1)}