Update README.md

README.md

@@ -9,7 +9,7 @@ The model was trained on around 300 FLUX images and 300 photographs from Unsplas
 Accuracy varies depending on the dataset but can be anywhere from 80%-90% depending on the dataset you use.
 
  Methods include fit, predict, score, and predict_proba. First, load an image using PIL (Pillow) and then store using an array. Load the class using joblib and then predict.
- The model is already pretrained but can be trained again using fit.
+ The model is already pretrained but can be trained again using fit. The images have to be resized to 512x512 before prediction, otherwise the model will give an error.
 
  ```python
 model = joblib.load("flux_classifier.pkl")