Update app.py
Browse files
app.py
CHANGED
|
@@ -46,8 +46,8 @@ def get_conversation_chain(vectorstore):
|
|
| 46 |
# Use HuggingFacePipeline with a smaller model like `t5-small`
|
| 47 |
t5_model = pipeline(
|
| 48 |
"text2text-generation",
|
| 49 |
-
model="
|
| 50 |
-
tokenizer="
|
| 51 |
max_new_tokens=512, # Increase the maximum token output
|
| 52 |
temperature=0.9,# Control creativity
|
| 53 |
#do_sample=True,
|
|
|
|
| 46 |
# Use HuggingFacePipeline with a smaller model like `t5-small`
|
| 47 |
t5_model = pipeline(
|
| 48 |
"text2text-generation",
|
| 49 |
+
model="google/flan-t5-base", # Smaller model for low-end systems
|
| 50 |
+
tokenizer="google/flan-t5-base",
|
| 51 |
max_new_tokens=512, # Increase the maximum token output
|
| 52 |
temperature=0.9,# Control creativity
|
| 53 |
#do_sample=True,
|