Vitrous commited on
Commit
7d5c2ac
·
verified ·
1 Parent(s): fd77735

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -63,7 +63,7 @@ def generate_response(prompt: str) -> str:
63
  top_k=40,
64
  repetition_penalty=1.1
65
  )
66
- generated_text = pipe(persona_prompt + user_prompt)[0]['generated_text']
67
 
68
  # Remove the "ASSISTANT:" prefix from the generated text
69
  assistant_response = generated_text.replace("ASSISTANT:", "").strip()
 
63
  top_k=40,
64
  repetition_penalty=1.1
65
  )
66
+ generated_text = pipe(prompt_template)[0]['generated_text']
67
 
68
  # Remove the "ASSISTANT:" prefix from the generated text
69
  assistant_response = generated_text.replace("ASSISTANT:", "").strip()