1NEYRON1 commited on
Commit
659f8c9
·
verified ·
1 Parent(s): 3805ab7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -141,6 +141,7 @@ id_to_cat = {0: 'High Energy Physics - Theory',
141
  137: 'Disordered Systems and Neural Networks'
142
  }
143
 
 
144
  def load_model():
145
  tokenizer = AutoTokenizer.from_pretrained('distilbert-base-cased')
146
  model = AutoModelForSequenceClassification.from_pretrained(
@@ -173,7 +174,7 @@ def classify_text(title, description):
173
  return [(label, score / total) for label, score in res]
174
 
175
  st.title("🔬 Классификация научных статей")
176
- st.markdown("Введите заголовок и краткое описание научной статьи, чтобы определить её тематические категории.")
177
 
178
  title = st.text_input("📝 Заголовок статьи", placeholder="Например: Deep Learning for Image Recognition")
179
  description = st.text_area("🧾 Краткое описание статьи", height=150, placeholder="Кратко опишите содержание статьи...")
 
141
  137: 'Disordered Systems and Neural Networks'
142
  }
143
 
144
+ @st.cache_resource
145
  def load_model():
146
  tokenizer = AutoTokenizer.from_pretrained('distilbert-base-cased')
147
  model = AutoModelForSequenceClassification.from_pretrained(
 
174
  return [(label, score / total) for label, score in res]
175
 
176
  st.title("🔬 Классификация научных статей")
177
+ st.markdown("Введите заголовок и краткое описание научной статьи, чтобы определить её тематические категории. (сервис работает только для англоязычных статей)")
178
 
179
  title = st.text_input("📝 Заголовок статьи", placeholder="Например: Deep Learning for Image Recognition")
180
  description = st.text_area("🧾 Краткое описание статьи", height=150, placeholder="Кратко опишите содержание статьи...")