jeevana commited on
Commit
52a0cab
·
verified ·
1 Parent(s): f77a27a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def generate_response(model, tokenizer, prompt):
27
  response = tokenizer.decode(output[0], skip_special_tokens=True)
28
  print("Generated response::", response )
29
  print("len(prompt)::", len(prompt) )
30
- response = response[len(prompt) + 8:]
31
  return response
32
 
33
  def predict(input):
 
27
  response = tokenizer.decode(output[0], skip_special_tokens=True)
28
  print("Generated response::", response )
29
  print("len(prompt)::", len(prompt) )
30
+ response = response[len(prompt) + 9:]
31
  return response
32
 
33
  def predict(input):