Update app.py

app.py

@@ -147,17 +147,22 @@ id_to_cat = {0: 'Performance',
  137: 'Artificial Intelligence'}
 
 # Загружаем модель (замените на вашу модель, если нужно)
-model_name = 'checkpoint'
-try:
+@st.cache_resource
+def load_model():
     tokenizer = AutoTokenizer.from_pretrained('distilbert-base-cased')
     model = AutoModelForSequenceClassification.from_pretrained(
-        model_name,
+        'checkpoint',
         num_labels=len(id_to_cat),
         problem_type="multi_label_classification"
     )
+    return model, tokenizer
+
+# Load model/tokenizer once and cache it
+try:
+    model, tokenizer = load_model()
 except OSError as e:
     st.error(f"Ошибка загрузки модели: {e}. Убедитесь, что модель доступна или укажите другую.")
-    st.stop()  # Остановка выполнения приложения при ошибке
+    st.stop()
 
 
 def classify_text(title, description):