Change model_base_id and lora_path

app.py

@@ -20,8 +20,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 from peft import PeftModel
 
 # === [1] Model and Tokenizer Loading ===
-base_model_id = "meta-llama/Llama-2-7b-hf"  # Replace with your base model
-lora_path = "./tat-llm"  # Path to your fine-tuned LoRA folder
+base_model_id = "NousResearch/Nous-Hermes-2-Mistral-7B-DPO"
+lora_path = "./tat-llm-final-e3"  # Path to your fine-tuned LoRA folder
 
 # Load base model and LoRA adapter
 base_model = AutoModelForCausalLM.from_pretrained(base_model_id, torch_dtype=torch.float16)