alex6095 commited on
Commit
620a618
·
1 Parent(s): 7de8e9d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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('monologg/distilkobert')
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)