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