Update app.py

app.py

@@ -9,7 +9,7 @@ from torch.utils.data import Dataset, DataLoader
 from transformers import RobertaModel, RobertaTokenizer
 from transformers import AutoModel, DistilBertTokenizer
 import transformers
-
+from transformers import pipeline
 
 idx_to_tag = {0: 'cs',
  1: 'stat',
@@ -32,7 +32,7 @@ tag_to_idx = {'cs': 0,
 
 tokenizer = DistilBertTokenizer.from_pretrained('distilbert-base-uncased')
 
-model = AutoModel.from_pretrained('./model', device_map='auto')
+model = pipeline('./model')
 
 
 st.markdown("### Угадыватель")