Update app.py
Browse files
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 |
-
|
| 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()
|