Soumen commited on
Commit
a2e9e23
·
1 Parent(s): b330918

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -99,6 +99,7 @@ def main():
99
  message = c1.text_input("Type your text here!")
100
  if st.button("CaptureImage"):
101
  camera_photo = c2.camera_input("Capture a photo to summarize: ", on_change=change_photo_state)
 
102
  if st.button("Stop camera"):
103
  CaptureImage =False
104
 
@@ -182,7 +183,7 @@ def main():
182
  return tokenizer, model
183
  if "history" not in st.session_state:
184
  st.session_state.history = []
185
- st.title("Hello bot: ")
186
  def generate_answer():
187
  tokenizer, model = get_models()
188
  user_message = st.session_state.input_text
@@ -193,12 +194,12 @@ def main():
193
  ) # .replace("<s>", "").replace("</s>", "")
194
  st.session_state.history.append({"message": user_message, "is_user": True})
195
  st.session_state.history.append({"message": message_bot, "is_user": False})
196
- st.text_input("Talk to the bot", key="input_text", on_change=generate_answer)
197
  from copyreg import clear_extension_cache
198
  for chat in st.session_state.history:
199
  st_message(**chat)
 
200
  if st.button("Refresh/New Chat"):
201
- st.session_state.history = []
202
 
203
  if __name__ == "__main__":
204
  main()
 
99
  message = c1.text_input("Type your text here!")
100
  if st.button("CaptureImage"):
101
  camera_photo = c2.camera_input("Capture a photo to summarize: ", on_change=change_photo_state)
102
+ st.image(camera_photo, caption='Uploaded Image', use_column_width=True)
103
  if st.button("Stop camera"):
104
  CaptureImage =False
105
 
 
183
  return tokenizer, model
184
  if "history" not in st.session_state:
185
  st.session_state.history = []
186
+ # st.title("Hello bot: ")
187
  def generate_answer():
188
  tokenizer, model = get_models()
189
  user_message = st.session_state.input_text
 
194
  ) # .replace("<s>", "").replace("</s>", "")
195
  st.session_state.history.append({"message": user_message, "is_user": True})
196
  st.session_state.history.append({"message": message_bot, "is_user": False})
 
197
  from copyreg import clear_extension_cache
198
  for chat in st.session_state.history:
199
  st_message(**chat)
200
+ st.text_input("Talk to the bot", key="input_text", on_change=generate_answer)
201
  if st.button("Refresh/New Chat"):
202
+ st.session_state.history = None
203
 
204
  if __name__ == "__main__":
205
  main()