m7mdal7aj commited on
Commit
5c3c21e
·
verified ·
1 Parent(s): 13ad7e8

Update my_model/state_manager.py

Browse files
Files changed (1) hide show
  1. my_model/state_manager.py +2 -3
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
- st.text("DDDDDDDDD")
296
  image_with_boxes, detected_objects_str = st.session_state['kbvqa'].detect_objects(img)
297
- st.text("TTTTTT")
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