prashant
commited on
Commit
·
9119fa1
1
Parent(s):
3f1c621
error fix
Browse files- 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
|
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)
|