Spaces:

vakodiya
/

streamlit-gpt2

Paused

vakodiya commited on Jul 17, 2024

Commit

5aefe30

verified ·

1 Parent(s): 3f30e93

Update generate_answer.py

Files changed (1) hide show

generate_answer.py CHANGED Viewed

@@ -1,8 +1,8 @@
-from transformers import GPT2Tokenizer, GPT2LMHeadModel
 from langchain.prompts import PromptTemplate
-tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
-model = GPT2LMHeadModel.from_pretrained('gpt2')
 def generate_answer(question):

+from transformers import AutoTokenizer, AutoModelForCausalLM
 from langchain.prompts import PromptTemplate
+model = AutoModelForCausalLM.from_pretrained("microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
 def generate_answer(question):