nico-che commited on
Commit
bc781bb
1 Parent(s): 2d0ef0f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -32,8 +32,8 @@ def client_generate(input, max_new_tokens=256, stop_sequences=[]):
32
  response = response[:response.find(seq)]
33
  return {'text': input, 'generated_text': response}
34
 
35
- def respond(message, chat_history, modelname=model_name, max_tokens=128):
36
- bot_message = client_generate(reshape_prompt(message, modelname),
37
  max_new_tokens=max_tokens,#1024,
38
  stop_sequences=["."], #stop_sequences to not generate the user answer
39
  )['generated_text']
 
32
  response = response[:response.find(seq)]
33
  return {'text': input, 'generated_text': response}
34
 
35
+ def respond(message, chat_history, max_tokens=128):
36
+ bot_message = client_generate(message,
37
  max_new_tokens=max_tokens,#1024,
38
  stop_sequences=["."], #stop_sequences to not generate the user answer
39
  )['generated_text']