Update rag.py

rag.py

@@ -55,8 +55,11 @@ class ChatPDF:
         # Intended Uses
         # Given the nature of the training data, the Phi-2 model is best suited for prompts using the 
         # QA format, the chat format, and the code format.
-        tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-2", trust_remote_code=True)
-        model = AutoModelForCausalLM.from_pretrained("microsoft/phi-2", trust_remote_code=True)
+        # tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-2", trust_remote_code=True)
+        # model = AutoModelForCausalLM.from_pretrained("microsoft/phi-2", trust_remote_code=True)
+        # https://huggingface.co/meta-llama/Llama-2-7b-chat-hf?library=true
+        tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-chat-hf")
+        model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-chat-hf")