Namitg02 commited on
Commit
08f351d
·
verified ·
1 Parent(s): 0bf4653

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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