Update app.py
Browse files
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("📊 Порог
|
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:
|