jonas commited on
Commit
4987d70
1 Parent(s): 00f3a9e

changed model path

Browse files
Files changed (1) hide show
  1. main_site.py +2 -1
main_site.py CHANGED
@@ -118,7 +118,8 @@ def app():
118
  # load from disc (github repo) for performance boost
119
  @st.cache(allow_output_mutation=True)
120
  def load_sdgClassifier():
121
- classifier = pipeline("text-classification", model= "../models/osdg_sdg/")
 
122
 
123
  return classifier
124
 
 
118
  # load from disc (github repo) for performance boost
119
  @st.cache(allow_output_mutation=True)
120
  def load_sdgClassifier():
121
+
122
+ classifier = pipeline("text-classification", model="jonas/sdg_classifier_osdg")
123
 
124
  return classifier
125