Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 = "
|
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)
|