zakerytclarke commited on
Commit
e886577
·
verified ·
1 Parent(s): 0b0a8ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -143,7 +143,7 @@ class TeapotAI:
143
  """
144
 
145
 
146
- result = self.generator(input_text)[0].get("generated_text")
147
 
148
 
149
  if self.settings.log_level == "debug":
@@ -335,11 +335,11 @@ def main():
335
 
336
  s1, s2, s3 = st.columns([1, 2, 3])
337
  with s1:
338
- suggestion_button("Tell me about the varieties of tea", teapot_ai)
339
  with s2:
340
  suggestion_button("Who was born first, Alan Turing or John von Neumann?", teapot_ai)
341
  with s3:
342
- suggestion_button("Extract Google's stock price", teapot_ai)
343
 
344
  if user_input:
345
  with st.chat_message("user"):
 
143
  """
144
 
145
 
146
+ result = self.generator(input_text, max_length=512)[0].get("generated_text")
147
 
148
 
149
  if self.settings.log_level == "debug":
 
335
 
336
  s1, s2, s3 = st.columns([1, 2, 3])
337
  with s1:
338
+ suggestion_button("Tell me about Earl Grey tea.", teapot_ai)
339
  with s2:
340
  suggestion_button("Who was born first, Alan Turing or John von Neumann?", teapot_ai)
341
  with s3:
342
+ suggestion_button("Extract the year Google was founded", teapot_ai)
343
 
344
  if user_input:
345
  with st.chat_message("user"):