PebinAPJ commited on
Commit
97cb255
·
verified ·
1 Parent(s): 1c43be6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -49,7 +49,8 @@ def get_conversation_chain(vectorstore):
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.7, # Control creativity
 
53
  top_p=0.9, # Nucleus sampling
54
  top_k=50,
55
  # Automatically fetches the token from Streamlit secrets
 
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.7,# Control creativity
53
+ do_sample=True,
54
  top_p=0.9, # Nucleus sampling
55
  top_k=50,
56
  # Automatically fetches the token from Streamlit secrets