update example

app.py

@@ -4,11 +4,11 @@ import streamlit as st
 from streamlit_image_select import image_select
 
 #Trick to not init function multitime
-# if "model" not in st.session_state:
-#     print("INIT MODEL")
-#     from src.model import Model
-#     st.session_state.model = Model()
-#     print("DONE INIT MODEL")
+if "model" not in st.session_state:
+    print("INIT MODEL")
+    from src.model import Model
+    st.session_state.model = Model()
+    print("DONE INIT MODEL")
 
 st.set_page_config(page_title="VQA", layout="wide")
 hide_menu_style = """
@@ -19,11 +19,11 @@ footer {visibility: hidden;}
 st.markdown(hide_menu_style, unsafe_allow_html= True)
 
 mapper = {
-        "images\\000000000008.jpg": "A",
-        "images\\000000000012.jpg": "B",
-        "images\\000000000016.jpg": "C",
-        "images\\000000000019.jpg": "D",
-        "images\\000000000181.jpg": "E"
+        "images/000000000645.jpg": "Đây là đâu",
+        "images/000000000661.jpg": "Tốc độ tối đa trên đoạn đường này là bao nhiêu",
+        "images/000000000674.jpg": "Còn bao xa nữa là tới Huế",
+        "images/000000000706.jpg": "Cầu này dài bao nhiêu",
+        "images/000000000777.jpg": "Chè khúc bạch giá bao nhiêu"
 }
 
 image = st.file_uploader("Choose an image file", type=["jpg", "jpeg", "png", "webp", ])
@@ -43,7 +43,7 @@ else:
 if 'image' in st.session_state:
     st.image(st.session_state.image)
     question = st.text_input("Question: ", value=st.session_state.question)
-#     if question:
-#         answer = st.session_state.model.inference(st.session_state.image, question)
-#         st.write(f"Answer: {answer}")
+    if question:
+        answer = st.session_state.model.inference(st.session_state.image, question)
+        st.write(f"Answer: {answer}")
         

requirements.txt

@@ -4,4 +4,5 @@ vietocr>=0.3.8
 pillow==9.5.0
 torchvision==0.18.0
 streamlit==1.35.0
-transformers==4.41.2
+transformers==4.41.2
+streamlit-image-select==0.6.0