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 |
|