KyanChen commited on
Commit
2fed23f
β€’
1 Parent(s): 3bd3201

Update pages/2_πŸ“·_A_Folder.py

Browse files
Files changed (1) hide show
  1. pages/2_πŸ“·_A_Folder.py +2 -2
pages/2_πŸ“·_A_Folder.py CHANGED
@@ -25,7 +25,7 @@ my_upload = st.sidebar.file_uploader("Upload multi images", type=["png", "jpg",
25
  col1, col2 = st.columns(2)
26
  parent_folder = './'
27
 
28
- @st.cache_resource
29
  def _init_model_return_results(imgs):
30
  cls_model = init_model(parent_folder + 'configs/resnet/resnet50_8xb32_in1k.py',
31
  'https://download.openmmlab.com/mmclassification/v0/resnet/resnet50_8xb32_in1k_20210831-ea4938fc.pth')
@@ -53,7 +53,7 @@ def _init_model_return_results(imgs):
53
  return results, class2idx
54
 
55
 
56
- @st.cache_data
57
  def _get_image(my_upload):
58
  if len(my_upload):
59
  img_files = my_upload
 
25
  col1, col2 = st.columns(2)
26
  parent_folder = './'
27
 
28
+ # @st.cache_resource
29
  def _init_model_return_results(imgs):
30
  cls_model = init_model(parent_folder + 'configs/resnet/resnet50_8xb32_in1k.py',
31
  'https://download.openmmlab.com/mmclassification/v0/resnet/resnet50_8xb32_in1k_20210831-ea4938fc.pth')
 
53
  return results, class2idx
54
 
55
 
56
+ # @st.cache_data
57
  def _get_image(my_upload):
58
  if len(my_upload):
59
  img_files = my_upload