Hugging Face's logo

Spaces:
ilushado
/
article_roberta_classifier
Runtime error

App Files Community
ilushado commited on
Commit
fcc8424
·
1 Parent(s): b42c3b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ tag_to_idx = {'cs': 0,
32
 
33
  tokenizer = DistilBertTokenizer.from_pretrained('distilbert-base-uncased')
34
 
35
- model = AutoModel.from_pretrained('./model')
36
 
37
 
38
  st.markdown("### Угадыватель")
 
32
 
33
  tokenizer = DistilBertTokenizer.from_pretrained('distilbert-base-uncased')
34
 
35
+ model = AutoModel.from_pretrained('./model', device_map='cpu')
36
 
37
 
38
  st.markdown("### Угадыватель")