from artifacts import crop_model_path, crop_transformer_path, crop_target_encoder_path | |
from artifacts import fertilizer_model_path, fertilizer_transformer_path, fertilizer_target_encoder_path | |
from artifacts import plant_diseases_classifier_model_path | |
from utils import load_model_and_encoders | |
from ultralytics import YOLO | |
crop_model, crop_pipeline_encoder, crop_label_encoder = load_model_and_encoders(model_path=crop_model_path, | |
transformer_path=crop_transformer_path, | |
target_encoder_path=crop_target_encoder_path) | |
fertilizer_model, fertilizer_pipeline_encoder, fertilizer_label_encoder = load_model_and_encoders(model_path=fertilizer_model_path, | |
transformer_path=fertilizer_transformer_path, | |
target_encoder_path=fertilizer_target_encoder_path) | |
plant_diseases_classifier_model = YOLO(plant_diseases_classifier_model_path) | |