Modified files for deployment

app.py

@@ -5,8 +5,12 @@ import logging
 import gc
 import warnings
 import os
+from huggingface_hub import login
 from config import MODEL_CONFIGS, DEFAULT_MODEL, MODEL_SETTINGS, GENERATION_DEFAULTS, MEDICAL_SYSTEM_PROMPT, UI_CONFIG
 
+# Login with the secret token
+login(token=os.getenv("HF_TOKEN"))
+
 # Suppress warnings
 warnings.filterwarnings("ignore")
 logging.getLogger("transformers").setLevel(logging.ERROR)

config.py

@@ -32,7 +32,7 @@ MODEL_CONFIGS = {
 }
 
 # Default model to use
-DEFAULT_MODEL = "medllama2"
+DEFAULT_MODEL = "meditron"
 
 # Model loading settings
 MODEL_SETTINGS = {
@@ -68,8 +68,8 @@ Remember: This information is for educational purposes only and should not repla
 
 # UI settings
 UI_CONFIG = {
-    "title": "🏥 MedLLaMA2 Medical Chatbot",
-    "description": "A medical AI assistant powered by MedLLaMA2. Please note: This is for educational purposes only and should not replace professional medical advice.",
+    "title": "🏥 Meditron Medical Chatbot",
+    "description": "A medical AI assistant powered by Meditron 7B. Please note: This is for educational purposes only and should not replace professional medical advice.",
     "examples": [
         "What are the symptoms of diabetes?",
         "How can I maintain a healthy heart?",