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