mathaisjustin's picture
Deploy Fish Disease Detection AI
fbbdeab
raw
history blame contribute delete
546 Bytes
"""
Backend package for Fish Disease Detection
Handles model loading, validation, prediction, and treatment recommendations
"""
from .config import (
CLASSES,
GEMINI_API_KEY,
GEMINI_MODEL_NAME,
MODEL_PATH,
CONFIDENCE_THRESHOLD,
DEVICE,
MAX_FILE_SIZE_MB,
MIN_IMAGE_SIZE_PX,
VALID_EXTENSIONS
)
from .predictor import FishDiseasePredictor
__version__ = "1.0.0"
__all__ = [
'FishDiseasePredictor',
'CLASSES',
'GEMINI_API_KEY',
'GEMINI_MODEL_NAME',
'MODEL_PATH',
'CONFIDENCE_THRESHOLD'
]