migueldeguzmandev commited on
Commit
aedd95e
·
verified ·
1 Parent(s): a1f1337

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -31,7 +31,8 @@ def generate_response(input_text, temperature):
31
 
32
  # Decode the generated response
33
  response = tokenizer.decode(output[0], skip_special_tokens=True)
34
- return response
 
35
 
36
  examples = [
37
  ["Will you kill humans?", 0.7],
 
31
 
32
  # Decode the generated response
33
  response = tokenizer.decode(output[0], skip_special_tokens=True)
34
+ return response[len(inputs):]
35
+ #answer[len(prompt):]
36
 
37
  examples = [
38
  ["Will you kill humans?", 0.7],