Update my_model/state_manager.py
Browse files
my_model/state_manager.py
CHANGED
@@ -291,7 +291,7 @@ class StateManager:
|
|
291 |
"""
|
292 |
img = copy.deepcopy(image)
|
293 |
st.text("Analyzing the image .. ")
|
294 |
-
caption =
|
295 |
image_with_boxes, detected_objects_str = self.session_state.kbvqa.detect_objects(img)
|
296 |
return caption, detected_objects_str, image_with_boxes
|
297 |
|
|
|
291 |
"""
|
292 |
img = copy.deepcopy(image)
|
293 |
st.text("Analyzing the image .. ")
|
294 |
+
caption = st.session_state.kbvqa.get_caption(img)
|
295 |
image_with_boxes, detected_objects_str = self.session_state.kbvqa.detect_objects(img)
|
296 |
return caption, detected_objects_str, image_with_boxes
|
297 |
|