Sam-Oliveira commited on
Commit
b662fa8
Β·
1 Parent(s): d0c1c6b

Change KeyBert

Browse files
Files changed (3) hide show
  1. src/db.py +2 -1
  2. src/scrape.py +1 -1
  3. src/streamlit_app.py +1 -2
src/db.py CHANGED
@@ -12,7 +12,8 @@ def get_conn():
12
  authors TEXT,
13
  abstract TEXT,
14
  published TEXT,
15
- summary TEXT
 
16
  )
17
  """
18
  )
 
12
  authors TEXT,
13
  abstract TEXT,
14
  published TEXT,
15
+ summary TEXT,
16
+ tags TEXT
17
  )
18
  """
19
  )
src/scrape.py CHANGED
@@ -4,7 +4,7 @@ from db import get_conn
4
  from config import MAX_RESULTS
5
  from keybert import KeyBERT
6
 
7
- _kw = KeyBERT("all-MiniLM-L6-v2") # ~90 MB, load once
8
 
9
  def make_tags(title, abstract, top_n=5):
10
  """
 
4
  from config import MAX_RESULTS
5
  from keybert import KeyBERT
6
 
7
+ _kw = KeyBERT()
8
 
9
  def make_tags(title, abstract, top_n=5):
10
  """
src/streamlit_app.py CHANGED
@@ -48,7 +48,7 @@ with tab1:
48
  ).fetchall()
49
  st.components.v1.html(render_rows(newest), height=600, scrolling=True)
50
 
51
- # ───────── TAB 2 – Digest (unchanged) ─────────────────────────────
52
  with tab2:
53
  st.header("Digest from local DB")
54
  d_topic = st.text_input("Keyword to match tags", value="large language")
@@ -57,7 +57,6 @@ with tab2:
57
  rows = rows_by_tag(d_topic, MAX_RESULTS)
58
  st.components.v1.html(render_rows(rows), height=800, scrolling=True)
59
 
60
- # ───────── TAB 3 – Ideate (new feedback) ──────────────────────────
61
  with tab3:
62
  st.header("Brainstorm new research ideas")
63
  mode = st.radio("Context source", ["Keyword", "ArXiv IDs"])
 
48
  ).fetchall()
49
  st.components.v1.html(render_rows(newest), height=600, scrolling=True)
50
 
51
+
52
  with tab2:
53
  st.header("Digest from local DB")
54
  d_topic = st.text_input("Keyword to match tags", value="large language")
 
57
  rows = rows_by_tag(d_topic, MAX_RESULTS)
58
  st.components.v1.html(render_rows(rows), height=800, scrolling=True)
59
 
 
60
  with tab3:
61
  st.header("Brainstorm new research ideas")
62
  mode = st.radio("Context source", ["Keyword", "ArXiv IDs"])