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