Update ME_LLAMA_MODEL to use the Llama-3.1-8B-Instruct variant and improve fallback model logging

medbot/config.py

@@ -1,2 +1,2 @@
-ME_LLAMA_MODEL = "clinicalnlplab/me-llama-13b"
+ME_LLAMA_MODEL = "meta-llama/Llama-3.1-8B-Instruct"
 FALLBACK_MODEL = "epfl-llm/meditron-7b" 

medbot/model.py

@@ -22,7 +22,7 @@ class ModelManager:
             )
         except Exception as e:
             print(f"Error loading model: {e}")
-            print("Falling back to Llama-2-7b-chat-hf...")
+            print(f"Falling back to {FALLBACK_MODEL}...")
             self.tokenizer = AutoTokenizer.from_pretrained(FALLBACK_MODEL)
             self.model = AutoModelForCausalLM.from_pretrained(
                 FALLBACK_MODEL,