Update app.py
Browse files
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="
|
53 |
temperature=0.5,
|
54 |
-
max_new_tokens=
|
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
|