m7mdal7aj commited on
Commit
1a4044e
1 Parent(s): df65239

Update my_model/state_manager.py

Browse files
Files changed (1) hide show
  1. my_model/state_manager.py +2 -2
my_model/state_manager.py CHANGED
@@ -260,7 +260,7 @@ class StateManager:
260
  Args:
261
  image_key (str): Unique key for the image.
262
  image (obj): The uploaded image data.
263
- kbvqa (KBVQA object): The loaded KBVQA model.
264
  """
265
 
266
  if image_key not in st.session_state['images_data']:
@@ -284,7 +284,7 @@ class StateManager:
284
 
285
  Args:
286
  image (obj): The image data to analyze.
287
- kbvqa (KBVQA object): The loaded KBVQA model.
288
 
289
  Returns:
290
  tuple: A tuple containing the generated caption, detected objects string, and image with bounding boxes.
 
260
  Args:
261
  image_key (str): Unique key for the image.
262
  image (obj): The uploaded image data.
263
+
264
  """
265
 
266
  if image_key not in st.session_state['images_data']:
 
284
 
285
  Args:
286
  image (obj): The image data to analyze.
287
+
288
 
289
  Returns:
290
  tuple: A tuple containing the generated caption, detected objects string, and image with bounding boxes.