final

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import torch
 from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
 import pickle
+import sklearn
+from sklearn.preprocessing import LabelEncoder
 
 # Скачать и загрузить токенизатор
 tokenizer = DistilBertTokenizer.from_pretrained('distilbert-base-cased')

requirements.txt

@@ -1,3 +1,4 @@
 transformers
 torch
 streamlit
+scikit-learn