vancauwe commited on
Commit
3b7d130
·
1 Parent(s): d136ce0

fix: disable caching of datasets on HF space

Browse files
Files changed (1) hide show
  1. src/main.py +3 -0
src/main.py CHANGED
@@ -13,6 +13,9 @@ from huggingface_hub import HfApi
13
  from transformers import pipeline
14
  from transformers import AutoModelForImageClassification
15
 
 
 
 
16
  import alps_map as sw_am
17
  import input_handling as sw_inp
18
  import obs_map as sw_map
 
13
  from transformers import pipeline
14
  from transformers import AutoModelForImageClassification
15
 
16
+ from datasets import disable_caching
17
+ disable_caching()
18
+
19
  import alps_map as sw_am
20
  import input_handling as sw_inp
21
  import obs_map as sw_map