PebinAPJ commited on
Commit
419b9d5
·
verified ·
1 Parent(s): f09be95

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -49,9 +49,9 @@ def get_vectorstore(text_chunks):
49
  def get_conversation_chain(vectorstore):
50
  # Use HuggingFaceEndpoint with explicitly passed parameters
51
  llm = HuggingFaceEndpoint(
52
- repo_id="google/flan-t5-base",
53
  temperature=0.5,
54
- max_new_tokens=248, # Explicit parameter
55
  )
56
 
57
  # Initialize memory
 
49
  def get_conversation_chain(vectorstore):
50
  # Use HuggingFaceEndpoint with explicitly passed parameters
51
  llm = HuggingFaceEndpoint(
52
+ repo_id="sshleifer/distilbart-cnn-12-6",
53
  temperature=0.5,
54
+ max_new_tokens=128, # Explicit parameter
55
  )
56
 
57
  # Initialize memory