Spaces:

GIZ
/

SDSN-demo

Running on CPU Upgrade

peter2000 commited on Aug 31, 2022

Commit

4b3b914

•

1 Parent(s): 2d65499

Update appStore/sdg_analysis.py

Files changed (1) hide show

appStore/sdg_analysis.py CHANGED Viewed

@@ -144,7 +144,7 @@ def app():
                 # load from disc (github repo) for performance boost
                 @st.cache(allow_output_mutation=True)
                 def load_sdgClassifier():
-                    classifier = pipeline("text-classification", model= "jonas/roberta-base-finetuned-sdg")
                     return classifier

                 # load from disc (github repo) for performance boost
                 @st.cache(allow_output_mutation=True)
                 def load_sdgClassifier():
+                    classifier = pipeline("text-classification", model= "jonas/sdg_classifier_osdg")
                     return classifier