update space

requirements.txt

@@ -14,3 +14,4 @@ numpy
 pillow
 huggingface_hub
 transformers
+keras

services/disease_detection_service.py

@@ -88,9 +88,6 @@ def load_llama_model():
         logger.error(f"Error loading Llama model: {e}", exc_info=True)
         raise RuntimeError("Failed to load Llama model") from e
 
-# Load models at startup
-disease_model = load_disease_model()
-llama_model, llama_tokenizer = load_llama_model()
 
 # Disease mapping and treatment guidelines
 # Maps predicted class indices to disease names, status, and treatment recommendations
@@ -186,5 +183,11 @@ class PoultryFarmBot:
         # Predict the disease using the disease detection model and return the response
         return self.predict_disease(image)
 
+# Load models at startup
+disease_model = load_disease_model()
+llama_model, llama_tokenizer = load_llama_model()
+
+
 # Initialize the bot instance
 bot = PoultryFarmBot(db)
+