File size: 1,039 Bytes
f45df09 |
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 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
"""
FarmEyes Models Package
=======================
AI model wrappers for the FarmEyes application.
Models:
- natlas_model: N-ATLaS hybrid model (API + GGUF) for translation and chat
- yolo_model: YOLOv11 for crop disease detection
"""
from models.natlas_model import (
NATLaSModel,
HuggingFaceAPIClient,
LocalGGUFModel,
get_natlas_model,
unload_natlas_model,
translate_text,
translate_batch,
LANGUAGE_NAMES,
NATIVE_LANGUAGE_NAMES
)
from models.yolo_model import (
YOLOModel,
PredictionResult,
get_yolo_model,
unload_yolo_model,
detect_disease,
detect_disease_with_image
)
__all__ = [
# N-ATLaS
"NATLaSModel",
"HuggingFaceAPIClient",
"LocalGGUFModel",
"get_natlas_model",
"unload_natlas_model",
"translate_text",
"translate_batch",
"LANGUAGE_NAMES",
"NATIVE_LANGUAGE_NAMES",
# YOLO
"YOLOModel",
"PredictionResult",
"get_yolo_model",
"unload_yolo_model",
"detect_disease",
"detect_disease_with_image"
]
|