Update my_model/state_manager.py

my_model/state_manager.py

@@ -260,7 +260,7 @@ class StateManager:
         else:
             return False  # No changes found   
 
-    def get_model(self) -> kbvqa:
+    def get_model(self) -> KBVQA:
         """
         Retrieves the KBVQA model from the session state.