migueldeguzmandev commited on
Commit
051b43a
1 Parent(s): 7cccd4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ 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
  # Create the Gradio interface
37
  interface = gr.Interface(
 
31
 
32
  # Decode the generated response
33
  response = tokenizer.decode(output[0], skip_special_tokens=True)
34
+ return response.replace(input_text, "").strip()
35
 
36
  # Create the Gradio interface
37
  interface = gr.Interface(