Update chatbot_bedrock.py
Browse filesremove hugging face token for model loading requirement
- chatbot_bedrock.py +3 -3
chatbot_bedrock.py
CHANGED
@@ -5,9 +5,9 @@ from langchain.chains import ConversationChain
|
|
5 |
import langchain.globals
|
6 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
7 |
|
8 |
-
def get_Model(
|
9 |
-
tokenizer = AutoTokenizer.from_pretrained("KvrParaskevi/Hotel-Assistant-Attempt4-Llama-2-7b"
|
10 |
-
model = AutoModelForCausalLM.from_pretrained("KvrParaskevi/Hotel-Assistant-Attempt4-Llama-2-7b"
|
11 |
return model
|
12 |
|
13 |
|
|
|
5 |
import langchain.globals
|
6 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
7 |
|
8 |
+
def get_Model():
|
9 |
+
tokenizer = AutoTokenizer.from_pretrained("KvrParaskevi/Hotel-Assistant-Attempt4-Llama-2-7b")
|
10 |
+
model = AutoModelForCausalLM.from_pretrained("KvrParaskevi/Hotel-Assistant-Attempt4-Llama-2-7b")
|
11 |
return model
|
12 |
|
13 |
|