Update README.md

README.md

@@ -14,8 +14,8 @@ This model predicts the Nitrogen, Phosphorus, and Potassium needs for crops base
 from transformers import AutoConfig, AutoModel
 
 # Load the model
-config = AutoConfig.from_pretrained("GodfreyOwino/NPK_prediction_model1")
-model = AutoModel.from_config(config)
+config = AutoConfig.from_pretrained("GodfreyOwino/NPK_prediction_model2", trust_remote_code=True)
+model = AutoModel.from_pretrained("GodfreyOwino/NPK_prediction_model2", config=config, trust_remote_code=True)
 
 input_data = {
     'crop_name': ['maize (corn)'],