Vitrous commited on
Commit
58a566f
·
verified ·
1 Parent(s): 043658f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -62,10 +62,10 @@ def generate_response(prompt: str) -> str:
62
  repetition_penalty=1.1
63
  )
64
  # Generate the response
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:
 
62
  repetition_penalty=1.1
63
  )
64
  # Generate the response
65
+ pipe_response = pipe(prompt_template + user_prompt)
66
+ generated_text = pipe_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: