Hugging Face's logo

Spaces:
Dzhamb
/
ArticleClassification
Runtime error

App Files Community
Dzhamb commited on
Commit
30bdc39
•
1 Parent(s): ab282f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ abstract = st.text_area("Введите аннотацию к статье, abst
16
  tokenizer = DistilBertTokenizerFast()
17
  model = DistilBertForSequenceClassification()
18
 
19
- load_model(model, 'weight_model')
20
  #model.load_state_dict(torch.load('weight_model'))
21
 
22
  text = get_text(title, abstract)
 
16
  tokenizer = DistilBertTokenizerFast()
17
  model = DistilBertForSequenceClassification()
18
 
19
+ model = load_model(model, 'weight_model')
20
  #model.load_state_dict(torch.load('weight_model'))
21
 
22
  text = get_text(title, abstract)