app/utils/config.py

import torch 
from pathlib import Path

DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'

BASE_DIR = Path(__file__).resolve().parent.parent 

FAISS_DIR = BASE_DIR / 'faiss_embedding'
MODELS_DIR = BASE_DIR / 'models'


BIOCLIP_MODEL_ID = 'imageomics/bioclip'
BIOCLIP_INDEX_PATH = FAISS_DIR / 'bioclip_animal_index.faiss'
BIOCLIP_EMBEDDINGS_PATH = FAISS_DIR / 'bioclip_animal_embeddings.pt'
ANIMAL_SPECIES_LIST = FAISS_DIR / 'animal_species_list.txt'
BIOCLIP_METADATA = FAISS_DIR / 'bioclip_metadata.json'


PLANT_MODEL_ID = 'drrobot9/BIONEXUS_PLANT_CLASSIFICATION'

QWEN_MODEL_ID = 'Qwen/Qwen2.5-1.5B-Instruct'

TOP_K_ANIMALS = 1
CONFIDENCE_THRESHOLD = 0.25