Spaces:
GIZ
/
Running on CPU Upgrade

prashant commited on
Commit
9119fa1
1 Parent(s): 3f1c621
Files changed (1) hide show
  1. utils/sdg_classifier.py +1 -1
utils/sdg_classifier.py CHANGED
@@ -52,7 +52,7 @@ def sdg_classification(haystackdoc:List[Document])->Tuple[DataFrame,Series]:
52
 
53
  if check_streamlit():
54
  st.write("caching model")
55
- classifier = st.cache(load_sdgClassifier(), allow_output_mutation=True)
56
  else:
57
  classifier = load_sdgClassifier()
58
  results = classifier.predict(haystackdoc)
 
52
 
53
  if check_streamlit():
54
  st.write("caching model")
55
+ classifier = st.cache(load_sdgClassifier, allow_output_mutation=True)
56
  else:
57
  classifier = load_sdgClassifier()
58
  results = classifier.predict(haystackdoc)