supernovamutinda
commited on
Commit
•
e64b0f5
1
Parent(s):
e90edfe
Update app.py
Browse files
app.py
CHANGED
@@ -20,6 +20,9 @@ if choice == "Meal Suggester":
|
|
20 |
# pip install accelerate
|
21 |
|
22 |
pipe = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-beta", torch_dtype=torch.bfloat16, device_map="auto")
|
|
|
|
|
|
|
23 |
|
24 |
# We use the tokenizer's chat template to format each message - see https://huggingface.co/docs/transformers/main/en/chat_templating
|
25 |
messages = [
|
|
|
20 |
# pip install accelerate
|
21 |
|
22 |
pipe = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-beta", torch_dtype=torch.bfloat16, device_map="auto")
|
23 |
+
|
24 |
+
user_input = st.input("insert text here", )
|
25 |
+
st.write(user_input)
|
26 |
|
27 |
# We use the tokenizer's chat template to format each message - see https://huggingface.co/docs/transformers/main/en/chat_templating
|
28 |
messages = [
|