Update app.py
Browse files
app.py
CHANGED
@@ -89,9 +89,9 @@ QA_CHAIN_PROMPT = PromptTemplate(input_variables=["context", "question"],templat
|
|
89 |
|
90 |
llm_model = "microsoft/Phi-3-mini-4k-instruct"
|
91 |
from transformers import AutoTokenizer
|
92 |
-
tokenizer = AutoTokenizer.from_pretrained(llm_model)
|
93 |
from transformers import AutoModelForCausalLM
|
94 |
-
model = AutoModelForCausalLM.from_pretrained(llm_model)
|
95 |
|
96 |
#question = "How can I reverse diabetes?"
|
97 |
|
|
|
89 |
|
90 |
llm_model = "microsoft/Phi-3-mini-4k-instruct"
|
91 |
from transformers import AutoTokenizer
|
92 |
+
tokenizer = AutoTokenizer.from_pretrained(llm_model,trust_remote_code=True)
|
93 |
from transformers import AutoModelForCausalLM
|
94 |
+
model = AutoModelForCausalLM.from_pretrained(llm_model,trust_remote_code=True)
|
95 |
|
96 |
#question = "How can I reverse diabetes?"
|
97 |
|