Update chatbot.py
Browse files- chatbot.py +1 -2
chatbot.py
CHANGED
@@ -52,7 +52,6 @@ def load_pipeline():
|
|
52 |
early_stopping=True,
|
53 |
num_beams = 2,
|
54 |
temperature = 0.1,
|
55 |
-
top_p = 0.8,
|
56 |
repetition_penalty = 1.03)
|
57 |
|
58 |
llm = HuggingFacePipeline(pipeline = pipe)
|
@@ -77,5 +76,5 @@ def demo_chain(input_text, memory):
|
|
77 |
memory=memory
|
78 |
)
|
79 |
|
80 |
-
chat_reply = conversation.invoke(input=input_text, max_new_tokens =
|
81 |
return chat_reply["response"]
|
|
|
52 |
early_stopping=True,
|
53 |
num_beams = 2,
|
54 |
temperature = 0.1,
|
|
|
55 |
repetition_penalty = 1.03)
|
56 |
|
57 |
llm = HuggingFacePipeline(pipeline = pipe)
|
|
|
76 |
memory=memory
|
77 |
)
|
78 |
|
79 |
+
chat_reply = conversation.invoke(input=input_text, max_new_tokens = 100)
|
80 |
return chat_reply["response"]
|