1NEYRON1 commited on
Commit
275db5d
·
1 Parent(s): 1a782a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -141,7 +141,7 @@ id_to_cat = {0: 'Performance',
141
  137: 'Artificial Intelligence'}
142
 
143
  # Загружаем модель (замените на вашу модель, если нужно)
144
- model_name = '1NEYRON1/Topic_classification_for_scientific_articles'
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(