Update app.py

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 import os
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer, AutoModel
 from huggingface_hub import login
 import torch
 
@@ -9,8 +9,8 @@ class VietnameseChatbot:
         """
         Initialize the Vietnamese chatbot with a pre-trained model
         """
-        self.tokenizer = AutoTokenizer.from_pretrained(model_name)
-        self.model = AutoModelForCausalLM.from_pretrained(model_name)
+        #self.tokenizer = AutoTokenizer.from_pretrained(model_name)
+        self.model = AutoModel.from_pretrained("TheBloke/Llama-2-7B-Chat-GGML")
         
         # Use GPU if available
         self.device = "cuda" if torch.cuda.is_available() else "cpu"