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

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[len(inputs):]
 
35
  #answer[len(prompt):]
36
 
37
  examples = [
 
31
 
32
  # Decode the generated response
33
  response = tokenizer.decode(output[0], skip_special_tokens=True)
34
+
35
+ return response.replace(input_text, "").strip()
36
  #answer[len(prompt):]
37
 
38
  examples = [