Spaces:

SanctiMoly
/

SanctiMolyTopic

Runtime error

alex6095 commited on Dec 12, 2021

Commit

43b149d

1 Parent(s): a9769aa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ tokenizer = KoBertTokenizer.from_pretrained('monologg/distilkobert')
 @st.cache(allow_output_mutation=True)
 def get_model():
-    model = DistilBertForSequenceClassification.from_pretrained('monologg/distilkobert', problem_type="multi_label_classification", num_labels=9)
     model.eval()
     return model

 @st.cache(allow_output_mutation=True)
 def get_model():
+    model = DistilBertForSequenceClassification.from_pretrained('alex6095/SanctiMolyTopic', problem_type="multi_label_classification", num_labels=9)
     model.eval()
     return model