|
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) |
|
|