1NEYRON1 commited on
Commit
4ca3015
·
1 Parent(s): e071fb5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -142,6 +142,7 @@ id_to_cat = {
142
  137: 'Algebraic Geometry'
143
  }
144
 
 
145
  def load_model():
146
  tokenizer = AutoTokenizer.from_pretrained('distilbert-base-cased')
147
  model = AutoModelForSequenceClassification.from_pretrained(
@@ -178,7 +179,7 @@ st.markdown("Введите заголовок и краткое описани
178
 
179
  title = st.text_input("📝 Заголовок статьи", placeholder="Например: Deep Learning for Image Recognition")
180
  description = st.text_area("🧾 Краткое описание статьи", height=150, placeholder="Кратко опишите содержание статьи...")
181
- top_percent = st.text_input("📊 Порог суммарной вероятности для предсказанных категорий (например, 95 или 0.95 для top 95%)", value="95")
182
 
183
  if st.button("🚀 Классифицировать"):
184
  if not title and not description:
 
142
  137: 'Algebraic Geometry'
143
  }
144
 
145
+ @st.cache_resource
146
  def load_model():
147
  tokenizer = AutoTokenizer.from_pretrained('distilbert-base-cased')
148
  model = AutoModelForSequenceClassification.from_pretrained(
 
179
 
180
  title = st.text_input("📝 Заголовок статьи", placeholder="Например: Deep Learning for Image Recognition")
181
  description = st.text_area("🧾 Краткое описание статьи", height=150, placeholder="Кратко опишите содержание статьи...")
182
+ top_percent = st.text_input("📊 Порог вероятности (например, 95 или 0.95 для top 95%)", value="95")
183
 
184
  if st.button("🚀 Классифицировать"):
185
  if not title and not description: