Update src/models.py

src/models.py

@@ -1,6 +1,21 @@
 from transformers import pipeline
 from langchain.llms import CTransformers
 from langchain.prompts import ChatPromptTemplate
+import os
+
+# Download models function
+def download_models():
+    """
+    Downloads the necessary models for the chatbot using wget. 
+    This function retrieves two models:
+    1. BioMistral-7B - A medical-specific language model.
+    2. Llama-2-7B - A general-purpose language model.
+    """
+    os.system('wget -O BioMistral-7B.Q4_K_M.gguf "https://huggingface.co/MaziyarPanahi/BioMistral-7B-GGUF/resolve/main/BioMistral-7B.Q4_K_M.gguf?download=true"')
+    os.system('wget -O llama-2-7b.Q4_K_M.gguf "https://huggingface.co/TheBloke/Llama-2-7B-GGUF/resolve/main/llama-2-7b.Q4_K_M.gguf?download=true"')
+
+# Call the download function to ensure the models are available before starting the bot
+download_models()
 
 # Text classification model for routing input
 classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")