Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ tokenizer = KoBertTokenizer.from_pretrained('monologg/distilkobert')
|
|
10 |
|
11 |
@st.cache(allow_output_mutation=True)
|
12 |
def get_model():
|
13 |
-
model = DistilBertForSequenceClassification.from_pretrained('
|
14 |
model.eval()
|
15 |
return model
|
16 |
|
|
|
10 |
|
11 |
@st.cache(allow_output_mutation=True)
|
12 |
def get_model():
|
13 |
+
model = DistilBertForSequenceClassification.from_pretrained('alex6095/SanctiMolyTopic', problem_type="multi_label_classification", num_labels=9)
|
14 |
model.eval()
|
15 |
return model
|
16 |
|