KyanChen commited on
Commit
3bd3201
Β·
1 Parent(s): 0458b98

Update pages/1_πŸ”₯_An_Image.py

Browse files
Files changed (1) hide show
  1. pages/1_πŸ”₯_An_Image.py +2 -2
pages/1_πŸ”₯_An_Image.py CHANGED
@@ -26,7 +26,7 @@ model_option = st.radio(
26
  parent_folder = './'
27
  topk = st.slider('Return top-k predictions', 1, 10, 3)
28
 
29
- @st.cache_resource
30
  def _init_model(model_option):
31
  if model_option == 'cls':
32
  # init model
@@ -44,7 +44,7 @@ def _init_model(model_option):
44
  visualizer = None
45
  return model, visualizer
46
 
47
- @st.cache_data
48
  def _get_image(my_upload=my_upload):
49
  if my_upload is not None:
50
  img_file = my_upload
 
26
  parent_folder = './'
27
  topk = st.slider('Return top-k predictions', 1, 10, 3)
28
 
29
+ # @st.cache_resource
30
  def _init_model(model_option):
31
  if model_option == 'cls':
32
  # init model
 
44
  visualizer = None
45
  return model, visualizer
46
 
47
+ # @st.cache_data
48
  def _get_image(my_upload=my_upload):
49
  if my_upload is not None:
50
  img_file = my_upload