yogl commited on
Commit
6b1ee78
·
verified ·
1 Parent(s): e16368c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -18,10 +18,10 @@ st.set_page_config(
18
  )
19
 
20
  HF_TOKEN = os.getenv("HF_TOKEN")
21
- HF_MERGED_REPO = os.getenv("HF_MERGED_REPO", "yogl/dissertations-merged")
22
- HF_EMB_REPO = os.getenv("HF_EMB_REPO", "yogl/rosrid-dissertations-embeddings")
23
 
24
- MODEL_NAME = os.getenv("MODEL_NAME", "intfloat/multilingual-e5-base")
25
 
26
  # Отображаемые заголовки (русские имена колонок)
27
  COLUMN_LABELS_RU = {
@@ -67,7 +67,7 @@ except Exception:
67
 
68
  @st.cache_data(show_spinner="Загрузка данных...")
69
  def load_data():
70
- # 1) МЕТА-ДАННЫЕ: yogl/dissertations-merged
71
  ds_meta = load_dataset(
72
  HF_MERGED_REPO,
73
  split="train",
@@ -77,7 +77,7 @@ def load_data():
77
  df_meta["registration_number"] = df_meta["registration_number"].astype(str)
78
  df_meta = df_meta.set_index("registration_number", drop=False)
79
 
80
- # 2) ЭМБЕДДИНГИ: yogl/rosrid-dissertations-embeddings
81
  ds_emb = load_dataset(
82
  HF_EMB_REPO,
83
  split="train",
 
18
  )
19
 
20
  HF_TOKEN = os.getenv("HF_TOKEN")
21
+ HF_MERGED_REPO = os.getenv("HF_MERGED_REPO")
22
+ HF_EMB_REPO = os.getenv("HF_EMB_REPO")
23
 
24
+ MODEL_NAME = os.getenv("MODEL_NAME")
25
 
26
  # Отображаемые заголовки (русские имена колонок)
27
  COLUMN_LABELS_RU = {
 
67
 
68
  @st.cache_data(show_spinner="Загрузка данных...")
69
  def load_data():
70
+ # 1) МЕТА-ДАННЫЕ
71
  ds_meta = load_dataset(
72
  HF_MERGED_REPO,
73
  split="train",
 
77
  df_meta["registration_number"] = df_meta["registration_number"].astype(str)
78
  df_meta = df_meta.set_index("registration_number", drop=False)
79
 
80
+ # 2) ЭМБЕДДИНГИ
81
  ds_emb = load_dataset(
82
  HF_EMB_REPO,
83
  split="train",