Vitrous commited on
Commit
043658f
·
verified ·
1 Parent(s): 7c6384b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -65,7 +65,7 @@ def generate_response(prompt: str) -> str:
65
  response = pipe(prompt_template + user_prompt)
66
  generated_text = response[0]['generated_text']
67
 
68
- return {"user": prompt, "assistant": generated_text
69
 
70
 
71
  def generate_prompt_response(persona_prompt: str, prompt: str) -> dict:
 
65
  response = pipe(prompt_template + user_prompt)
66
  generated_text = response[0]['generated_text']
67
 
68
+ return {"user": prompt, "assistant": generated_text
69
 
70
 
71
  def generate_prompt_response(persona_prompt: str, prompt: str) -> dict: