Update app.py

app.py

@@ -9,7 +9,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
 import torch
 
 # Load model and tokenizer
-MODEL_NAME = "vi-gemma2-2b-ChatQA-RAG"  # Thay thế bằng tên mô hình thực tế của bạn
+MODEL_NAME = "lora_model"  # Thay thế bằng tên mô hình thực tế của bạn
 model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, torch_dtype=torch.float16, device_map="auto")
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)