Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -15,14 +15,14 @@ def main():
|
|
15 |
|
16 |
# Chat interface
|
17 |
st.markdown("---")
|
18 |
-
st.header("
|
19 |
|
20 |
# Input for user-provided message
|
21 |
user_message = st.text_area("Enter your message:")
|
22 |
|
23 |
-
if st.button("Generate
|
24 |
if user_message.strip():
|
25 |
-
with st.spinner("Generating
|
26 |
# Load the pipeline
|
27 |
pipe = pipeline("text-generation", model="TinyLlama/TinyLlama-1.1B-Chat-v1.0", torch_dtype=torch.bfloat16, device_map="auto")
|
28 |
|
|
|
15 |
|
16 |
# Chat interface
|
17 |
st.markdown("---")
|
18 |
+
st.header("Nudge Response")
|
19 |
|
20 |
# Input for user-provided message
|
21 |
user_message = st.text_area("Enter your message:")
|
22 |
|
23 |
+
if st.button("Generate Nudge"):
|
24 |
if user_message.strip():
|
25 |
+
with st.spinner("Generating Nudge..."):
|
26 |
# Load the pipeline
|
27 |
pipe = pipeline("text-generation", model="TinyLlama/TinyLlama-1.1B-Chat-v1.0", torch_dtype=torch.bfloat16, device_map="auto")
|
28 |
|