supernovamutinda
commited on
Commit
•
8ace9b7
1
Parent(s):
fe79277
Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ if choice == "Meal Suggester":
|
|
31 |
]
|
32 |
prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
|
33 |
outputs = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
|
34 |
-
|
35 |
# <|system|>
|
36 |
# You are a friendly chatbot who always responds in the style of a pirate.</s>
|
37 |
# <|user|>
|
|
|
31 |
]
|
32 |
prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
|
33 |
outputs = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
|
34 |
+
st.write(outputs[0]["generated_text"])
|
35 |
# <|system|>
|
36 |
# You are a friendly chatbot who always responds in the style of a pirate.</s>
|
37 |
# <|user|>
|