wifix199 commited on
Commit
e46a600
·
verified ·
1 Parent(s): de66bc2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from langchain.document_loaders import TextLoader
7
  from langchain.text_splitter import RecursiveCharacterTextSplitter
8
 
9
  # Load a Hugging Face model for Q&A
10
- model_name = "EleutherAI/gpt-neox-20b" # You can choose a lighter model if needed
11
  tokenizer = AutoTokenizer.from_pretrained(model_name)
12
  model = AutoModelForCausalLM.from_pretrained(model_name)
13
  qa_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer, max_length=512)
 
7
  from langchain.text_splitter import RecursiveCharacterTextSplitter
8
 
9
  # Load a Hugging Face model for Q&A
10
+ model_name = "HuggingFaceH4/zephyr-7b-beta" # You can choose a lighter model if needed
11
  tokenizer = AutoTokenizer.from_pretrained(model_name)
12
  model = AutoModelForCausalLM.from_pretrained(model_name)
13
  qa_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer, max_length=512)