File size: 546 Bytes
fbbdeab
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
"""
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'
]