Update my_model/state_manager.py
Browse files
my_model/state_manager.py
CHANGED
@@ -260,7 +260,7 @@ class StateManager:
|
|
260 |
else:
|
261 |
return False # No changes found
|
262 |
|
263 |
-
def get_model(self) ->
|
264 |
"""
|
265 |
Retrieves the KBVQA model from the session state.
|
266 |
|
|
|
260 |
else:
|
261 |
return False # No changes found
|
262 |
|
263 |
+
def get_model(self) -> kbvqa:
|
264 |
"""
|
265 |
Retrieves the KBVQA model from the session state.
|
266 |
|