Update app.py
Browse files
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 |
-
|
|
|
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:
|