Spaces:
Sleeping
Sleeping
Update chatbot.py
Browse files- chatbot.py +2 -2
chatbot.py
CHANGED
@@ -48,8 +48,8 @@ def create_chain(chains, pdf_doc):
|
|
48 |
|
49 |
def create_model():
|
50 |
hf_api_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
|
51 |
-
tokenizer = AutoTokenizer.from_pretrained("google/
|
52 |
-
model = AutoModelForCausalLM.from_pretrained("google/
|
53 |
device_map='auto',
|
54 |
torch_dtype=torch.float16,
|
55 |
token=hf_api_token
|
|
|
48 |
|
49 |
def create_model():
|
50 |
hf_api_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
|
51 |
+
tokenizer = AutoTokenizer.from_pretrained("google-bert/bert-large-uncased-whole-word-masking-finetuned-squad")
|
52 |
+
model = AutoModelForCausalLM.from_pretrained("google-bert/bert-large-uncased-whole-word-masking-finetuned-squad",
|
53 |
device_map='auto',
|
54 |
torch_dtype=torch.float16,
|
55 |
token=hf_api_token
|