CoffeBank commited on
Commit
14b4992
·
1 Parent(s): a52adbc
Files changed (1) hide show
  1. demo/binary_classifier_demo.py +5 -4
demo/binary_classifier_demo.py CHANGED
@@ -10,8 +10,7 @@ from model_utils import load_model, load_ternary_model, classify_text
10
  from binoculars_utils import compute_scores, cleanup_model, cleanup_models
11
 
12
  MINIMUM_TOKENS = 1000
13
-
14
- SAMPLE_TEXT = """Привет!\tЯ хотел бы поделиться с вами своим опытом путешествия по Санкт-Петербургу — одному из самых красивых и удивительных городов России.\tЭто было по-настоящему незабываемое приключение, которое оставило множество ярких впечатлений.\tСанкт-Петербург сразу поразил меня своей атмосферой.\tГород буквально дышит историей: старинные здания, величественные соборы, широкие проспекты и, конечно же, каналы — всё это создает уникальный облик.\tПрогуливаясь по улицам, я чувствовал себя так, будто перенёсся в другое время, в эпоху империи, великих художников, писателей и архитекторов.\tОсобенно яркое впечатление на меня произвёл Эрмитаж.\tЭто не просто музей — это настоящий дворец искусства.\tОгромное здание Зимнего дворца снаружи выглядит торжественно и величественно, а внутри скрывает одну из крупнейших коллекций произведений искусства в мире.\tЯ провёл там несколько часов, но мне показалось, что этого времени всё равно недостаточно, чтобы увидеть и осмыслить всё, что там представлено.\tКартины, скульптуры, антикварная мебель, уникальные экспозиции — всё это завораживало и вдохновляло.\tНе менее захватывающим оказалось путешествие по каналам города.\tСанкт-Петербург не зря называют «Северной Венецией»: реки и каналы буквально пронизывают его карту, а прогулка на теплоходе позволила взглянуть на город с нового, водного ракурса.\tЯ любовался старинными мостами — каждый со своей историей и архитектурным стилем — и фасадами домов, отражающимися в воде.\tЭто было невероятно романтично и красиво, особенно когда солнце начинало садиться, заливая город золотистым светом.\tВ целом, поездка в Петербург стала для меня настоящим открытием.\tЭтот город невозможно охватить за одно путешествие — он требует времени, внимания и желания узнавать всё больше.\tЯ бы с радостью вернулся туда снова, чтобы ещё глубже погрузиться в его атмосферу, посетить новые музеи, открыть для себя скрытые уголки и просто снова почувствовать эту магию, которую излучает Петербург."""
15
 
16
  css = """
17
  .human-text {
@@ -106,8 +105,6 @@ def run_classifier(text, mode="binary", show_analysis=False):
106
 
107
  ### Вероятности классов:
108
  {prob_str}
109
-
110
- {scores_str}
111
  """
112
 
113
  # Analysis markdown
@@ -165,6 +162,10 @@ def run_classifier(text, mode="binary", show_analysis=False):
165
 
166
  analysis_md = "## Анализ текста\n\n"
167
 
 
 
 
 
168
  # Basic statistics
169
  analysis_md += "### Основная статистика\n"
170
  for key, value in text_analysis.get('basic_stats', {}).items():
 
10
  from binoculars_utils import compute_scores, cleanup_model, cleanup_models
11
 
12
  MINIMUM_TOKENS = 1000
13
+ SAMPLE_TEXT = """Привет! Я хотел бы поделиться с вами своим опытом путешествия по Санкт-Петербургу — одному из самых красивых и удивительных городов России. Это было по-настоящему незабываемое приключение, которое оставило множество ярких впечатлений. Санкт-Петербург сразу поразил меня своей атмосферой. Город буквально дышит историей: старинные здания, величественные соборы, широкие проспекты и, конечно же, каналы — всё это создает уникальный облик. Прогуливаясь по улицам, я чувствовал себя так, будто перенёсся в другое время, в эпоху империи, великих художников, писателей и архитекторов. Особенно яркое впечатление на меня произвёл Эрмитаж. Это не просто музей — это настоящий дворец искусства. Огромное з��ание Зимнего дворца снаружи выглядит торжественно и величественно, а внутри скрывает одну из крупнейших коллекций произведений искусства в мире. Я провёл там несколько часов, но мне показалось, что этого времени всё равно недостаточно, чтобы увидеть и осмыслить всё, что там представлено. Картины, скульптуры, антикварная мебель, уникальные экспозиции — всё это завораживало и вдохновляло. Не менее захватывающим оказалось путешествие по каналам города. Санкт-Петербург не зря называют «Северной Венецией»: реки и каналы буквально пронизывают его карту, а прогулка на теплоходе позволила взглянуть на город с нового, водного ракурса. Я любовался старинными мостами — каждый со своей историей и архитектурным стилем — и фасадами домов, отражающимися в воде. Это было невероятно романтично и красиво, особенно когда солнце начинало садиться, заливая город золотистым светом. В целом, поездка в Петербург стала для меня настоящим открытием. Этот город невозможно охватить за одно путешествие — он требует времени, внимания и желания узнавать всё больше. Я бы с радостью вернулся туда снова, чтобы ещё глубже погрузиться в его атмосферу, посетить новые музеи, открыть для себя скрытые уголки и просто снова почувствовать эту магию, которую излучает Петербург."""
 
14
 
15
  css = """
16
  .human-text {
 
105
 
106
  ### Вероятности классов:
107
  {prob_str}
 
 
108
  """
109
 
110
  # Analysis markdown
 
162
 
163
  analysis_md = "## Анализ текста\n\n"
164
 
165
+ # Add Binoculars Scores to analysis section
166
+ if scores:
167
+ analysis_md += scores_str + "\n"
168
+
169
  # Basic statistics
170
  analysis_md += "### Основная статистика\n"
171
  for key, value in text_analysis.get('basic_stats', {}).items():