Vineedhar commited on
Commit
075fa02
·
verified ·
1 Parent(s): 2a2ea30

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -15,14 +15,14 @@ def main():
15
 
16
  # Chat interface
17
  st.markdown("---")
18
- st.header("Chat Interface")
19
 
20
  # Input for user-provided message
21
  user_message = st.text_area("Enter your message:")
22
 
23
- if st.button("Generate Response"):
24
  if user_message.strip():
25
- with st.spinner("Generating response..."):
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