WICKED4950 commited on
Commit
c538203
·
verified ·
1 Parent(s): fe0a282

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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|>", "")