Update app.py
Browse files
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 |
-
|
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:
|