PebinAPJ commited on
Commit
c7e2211
·
verified ·
1 Parent(s): b65f6ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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="meta-llama/Meta-Llama-3-8B", # Smaller model for low-end systems
50
- tokenizer="meta-llama/Meta-Llama-3-8B",
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,