Update app.py
Browse files
app.py
CHANGED
@@ -62,10 +62,10 @@ def generate_response(prompt: str) -> str:
|
|
62 |
repetition_penalty=1.1
|
63 |
)
|
64 |
# Generate the response
|
65 |
-
|
66 |
-
generated_text =
|
67 |
|
68 |
-
|
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:
|