KvrParaskevi commited on
Commit
b4c5c5b
·
verified ·
1 Parent(s): 22ad1b8

Update chatbot.py

Browse files
Files changed (1) hide show
  1. 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 = 20)
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"]