davertor commited on
Commit
7eda7b6
β€’
1 Parent(s): 336e50a

debug error

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -218,12 +218,15 @@ st_color_option = st.sidebar.selectbox('Select colorizer mode',
218
  # det_nms_thres = st.sidebar.slider("Non-maximum supression IoU", 0.1, 0.9, value=0.4, step=0.1)
219
 
220
  # Load models
 
221
  try:
222
  colorizer = load_model(model_folder, st_color_option)
223
  except:
224
  pass
225
  colorizer = True
226
  # print('Error while loading the model. Please refresh the page')
 
 
227
 
228
  if colorizer is not None:
229
  st_title_message.markdown("**To begin, please upload an image** πŸ‘‡")
 
218
  # det_nms_thres = st.sidebar.slider("Non-maximum supression IoU", 0.1, 0.9, value=0.4, step=0.1)
219
 
220
  # Load models
221
+ print('before loading the model')
222
  try:
223
  colorizer = load_model(model_folder, st_color_option)
224
  except:
225
  pass
226
  colorizer = True
227
  # print('Error while loading the model. Please refresh the page')
228
+ print('after loading the model')
229
+
230
 
231
  if colorizer is not None:
232
  st_title_message.markdown("**To begin, please upload an image** πŸ‘‡")