Manofem commited on
Commit
1e607af
·
verified ·
1 Parent(s): 55bafcf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,7 +16,8 @@ def correct_words(text):
16
  def generate_response(message):
17
  global history
18
  for _ in range(2): # Assuming you want two responses, adjust as needed
19
- response = llm.generate(tokens=message.split(), top_k=50, top_p=0.95, temperature=1.0, repetition_penalty=1.0, last_n_tokens=1)
 
20
  response_text = ' '.join(map(str, response))
21
  time.sleep(2)
22
  corrected_response = correct_words(response_text)
@@ -24,7 +25,6 @@ def generate_response(message):
24
  yield ' '.join(history)
25
  # Clear the history list after the last response
26
  history = ["Chatbot:"]
27
-
28
  def chatbot(message, history):
29
  response_generator = generate_response(message)
30
  for response in response_generator:
 
16
  def generate_response(message):
17
  global history
18
  for _ in range(2): # Assuming you want two responses, adjust as needed
19
+ tokens = message.split()
20
+ response = llm.generate(tokens=tokens, top_k=50, top_p=0.95, temperature=1.0, repetition_penalty=1.0, last_n_tokens=1)
21
  response_text = ' '.join(map(str, response))
22
  time.sleep(2)
23
  corrected_response = correct_words(response_text)
 
25
  yield ' '.join(history)
26
  # Clear the history list after the last response
27
  history = ["Chatbot:"]
 
28
  def chatbot(message, history):
29
  response_generator = generate_response(message)
30
  for response in response_generator: