Spaces:
Sleeping
Sleeping
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 |
|