KvrParaskevi commited on
Commit
90a40c9
1 Parent(s): fd2ac6f

Update chatbot_bedrock.py

Browse files

remove hugging face token for model loading requirement

Files changed (1) hide show
  1. 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(hugging_face_key):
9
- tokenizer = AutoTokenizer.from_pretrained("KvrParaskevi/Hotel-Assistant-Attempt4-Llama-2-7b",use_auth_token=hugging_face_key)
10
- model = AutoModelForCausalLM.from_pretrained("KvrParaskevi/Hotel-Assistant-Attempt4-Llama-2-7b",use_auth_token=hugging_face_key).eval()
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