Update app.py
Browse files
app.py
CHANGED
@@ -141,7 +141,7 @@ id_to_cat = {0: 'Performance',
|
|
141 |
137: 'Artificial Intelligence'}
|
142 |
|
143 |
# Загружаем модель (замените на вашу модель, если нужно)
|
144 |
-
model_name = '
|
145 |
try:
|
146 |
tokenizer = AutoTokenizer.from_pretrained('distilbert-base-cased')
|
147 |
model = AutoModelForSequenceClassification.from_pretrained(
|
|
|
141 |
137: 'Artificial Intelligence'}
|
142 |
|
143 |
# Загружаем модель (замените на вашу модель, если нужно)
|
144 |
+
model_name = 'checkpoint'
|
145 |
try:
|
146 |
tokenizer = AutoTokenizer.from_pretrained('distilbert-base-cased')
|
147 |
model = AutoModelForSequenceClassification.from_pretrained(
|