Update app.py

app.py

@@ -4,7 +4,7 @@ model = keras.models.load_model('potatoes2.h5')
 class_names = ['Potato___Early_blight', 'Potato___Late_blight', 'Potato___healthy']
 
 def predict_input_image(img):
-  img_4d=img.reshape(-1,256,256,1)
+  img_4d=img.reshape(-1,256,256,3)
   prediction=model.predict(img_4d)[0]
   return {class_names[i]: float(prediction[i]) for i in range(len(class_names))}