Update chatbot.py

chatbot.py

@@ -48,8 +48,8 @@ def create_chain(chains, pdf_doc):
 
 def create_model():
     hf_api_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
-    tokenizer = AutoTokenizer.from_pretrained("google-bert/bert-large-uncased-whole-word-masking-finetuned-squad")
-    model = AutoModelForCausalLM.from_pretrained("google-bert/bert-large-uncased-whole-word-masking-finetuned-squad",
+    tokenizer = AutoTokenizer.from_pretrained("OpenAssistant/oasst-sft-1-pythia-12b")
+    model = AutoModelForCausalLM.from_pretrained("OpenAssistant/oasst-sft-1-pythia-12b",
                                                 device_map='auto',
                                                 torch_dtype=torch.float16,
                                                 token=hf_api_token