CR7CAD commited on
Commit
12cd000
·
verified ·
1 Parent(s): 4315c14

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def load_models():
30
  with st.spinner("Loading AI models... This may take a minute on first run."):
31
  models = {}
32
  # Load summarization model
33
- models['summarizer'] = pipeline("summarization", model="MikaSie/BART_no_extraction_V2")
34
 
35
  # Load feature extraction model for similarity
36
  models['feature_extractor'] = pipeline("feature-extraction", model="bert-base-uncased")
 
30
  with st.spinner("Loading AI models... This may take a minute on first run."):
31
  models = {}
32
  # Load summarization model
33
+ models['summarizer'] = pipeline("summarization", model="utrobinmv/t5_summary_en_ru_zh_base_2048")
34
 
35
  # Load feature extraction model for similarity
36
  models['feature_extractor'] = pipeline("feature-extraction", model="bert-base-uncased")