elozano commited on
Commit
81a3cd1
1 Parent(s): a59a2c1

Model names updated

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -15,9 +15,9 @@ ENTITY_COLOR = {
15
 
16
  def run() -> None:
17
  analyzer = NewsAnalyzer(
18
- category_model_name="elozano/news-category",
19
- fake_model_name="elozano/news-fake",
20
- clickbait_model_name="elozano/news-clickbait",
21
  ner_model_name="dslim/bert-base-NER",
22
  )
23
  st.title("📰 News Analyzer")
 
15
 
16
  def run() -> None:
17
  analyzer = NewsAnalyzer(
18
+ category_model_name="elozano/bert-base-cased-news-category",
19
+ fake_model_name="elozano/bert-base-cased-fake-news",
20
+ clickbait_model_name="elozano/bert-base-cased-clickbait-news",
21
  ner_model_name="dslim/bert-base-NER",
22
  )
23
  st.title("📰 News Analyzer")