Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -66,11 +66,13 @@ def respond(message, history):
|
|
66 |
for chunk in history:
|
67 |
give_mod += f"<|SOH|>{chunk[0]}<|SOB|>{chunk[1]}"
|
68 |
give_mod += f"<|SOH|>{message.capitalize()}<|SOB|>"
|
|
|
69 |
|
70 |
response = ""
|
71 |
for token in raw_pred(give_mod, model, tokenizer):
|
72 |
if "<|Clean|>" in token:
|
73 |
response = token
|
|
|
74 |
else:
|
75 |
response += token
|
76 |
yield response.replace("<|SOH|>", "").replace("<|Clean|>", "")
|
|
|
66 |
for chunk in history:
|
67 |
give_mod += f"<|SOH|>{chunk[0]}<|SOB|>{chunk[1]}"
|
68 |
give_mod += f"<|SOH|>{message.capitalize()}<|SOB|>"
|
69 |
+
print(give_mod)
|
70 |
|
71 |
response = ""
|
72 |
for token in raw_pred(give_mod, model, tokenizer):
|
73 |
if "<|Clean|>" in token:
|
74 |
response = token
|
75 |
+
print(response)
|
76 |
else:
|
77 |
response += token
|
78 |
yield response.replace("<|SOH|>", "").replace("<|Clean|>", "")
|