Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -46,7 +46,7 @@ class SanctiMoly(nn.Module):
|
|
46 |
|
47 |
@st.cache(allow_output_mutation=True)
|
48 |
def get_model():
|
49 |
-
bert_model = DistilBertModel.from_pretrained('
|
50 |
tokenizer = KoBertTokenizer.from_pretrained('monologg/distilkobert')
|
51 |
|
52 |
model = SanctiMoly(bert_model, freeze_bert=False)
|
|
|
46 |
|
47 |
@st.cache(allow_output_mutation=True)
|
48 |
def get_model():
|
49 |
+
bert_model = DistilBertModel.from_pretrained('alex6095/SanctiMolyOH_Cpu')
|
50 |
tokenizer = KoBertTokenizer.from_pretrained('monologg/distilkobert')
|
51 |
|
52 |
model = SanctiMoly(bert_model, freeze_bert=False)
|