Update my_model/state_manager.py

my_model/state_manager.py

@@ -7,7 +7,7 @@ import pandas as pd
 import copy
 import time
 from PIL import Image
-from typing import Tuple, Dict
+from typing import Tuple, Dict, Optional
 import streamlit as st
 from my_model.utilities.gen_utilities import free_gpu_resources
 from my_model.KBVQA import KBVQA, prepare_kbvqa_model