Spaces:
Sleeping
Sleeping
Update my_model/state_manager.py
Browse files
my_model/state_manager.py
CHANGED
|
@@ -105,7 +105,6 @@ class StateManager:
|
|
| 105 |
self.col3.write("##### Current Model Settings:")
|
| 106 |
data = [{'Setting': key, 'Value': str(value)} for key, value in st.session_state.items() if key in ["confidence_level", 'detection_model', 'method', 'kbvqa', 'previous_state', 'settings_changed', 'loading_in_progress', 'model_loaded', 'time_taken_to_load_model' ]]
|
| 107 |
df = pd.DataFrame(data).reset_index(drop=True)
|
| 108 |
-
#styled_df = df.style.set_properties(**{'background-color': 'white', 'color': 'black', 'border-color': 'black'}).set_table_styles([{'selector': 'th','props': [('background-color', 'gray'), ('font-weight', 'bold')]}])
|
| 109 |
return self.col3.write(df)
|
| 110 |
|
| 111 |
|
|
@@ -292,9 +291,9 @@ class StateManager:
|
|
| 292 |
img = copy.deepcopy(image)
|
| 293 |
st.text("Analyzing the image .. ")
|
| 294 |
caption = st.session_state['kbvqa'].get_caption(img)
|
| 295 |
-
|
| 296 |
image_with_boxes, detected_objects_str = st.session_state['kbvqa'].detect_objects(img)
|
| 297 |
-
|
| 298 |
return caption, detected_objects_str, image_with_boxes
|
| 299 |
|
| 300 |
|
|
|
|
| 105 |
self.col3.write("##### Current Model Settings:")
|
| 106 |
data = [{'Setting': key, 'Value': str(value)} for key, value in st.session_state.items() if key in ["confidence_level", 'detection_model', 'method', 'kbvqa', 'previous_state', 'settings_changed', 'loading_in_progress', 'model_loaded', 'time_taken_to_load_model' ]]
|
| 107 |
df = pd.DataFrame(data).reset_index(drop=True)
|
|
|
|
| 108 |
return self.col3.write(df)
|
| 109 |
|
| 110 |
|
|
|
|
| 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
|
| 298 |
|
| 299 |
|