davidberenstein1957 HF staff commited on
Commit
b82c93a
·
verified ·
1 Parent(s): 581cf0b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -7,8 +7,8 @@ from huggingface_hub import get_token
7
 
8
  static_embedding = StaticEmbedding.from_model2vec("minishlab/potion-base-8M")
9
  model = SentenceTransformer(modules=[static_embedding])
10
- dataset_name = "smol-blueprint/fineweb-bbc-news-text-embeddings"
11
- embedding_column = "embedding"
12
  table_name = "fineweb"
13
 
14
  duckdb.sql(query=f"""
@@ -34,7 +34,7 @@ def similarity_search(query: str, k: int = 5):
34
  with gr.Blocks() as demo:
35
  gr.Markdown("""# RAG - retrieve
36
 
37
- Part of [smol blueprint](https://github.com/huggingface/smol-blueprint) - a smol blueprint for AI development, focusing on applied examples of RAG, information extraction, analysis and fine-tuning in the age of LLMs. """)
38
  query = gr.Textbox(label="Query")
39
  k = gr.Slider(1, 50, value=5, label="Number of results")
40
  btn = gr.Button("Search")
 
7
 
8
  static_embedding = StaticEmbedding.from_model2vec("minishlab/potion-base-8M")
9
  model = SentenceTransformer(modules=[static_embedding])
10
+ dataset_name = "ai-blueprint/fineweb-bbc-news-embeddings"
11
+ embedding_column = "embeddings"
12
  table_name = "fineweb"
13
 
14
  duckdb.sql(query=f"""
 
34
  with gr.Blocks() as demo:
35
  gr.Markdown("""# RAG - retrieve
36
 
37
+ Part of [AI blueprint](https://github.com/davidberenstein1957/ai-blueprint) - a blueprint for AI development, focusing on practical examples of RAG, information extraction, analysis and fine-tuning in the age of LLMs. """)
38
  query = gr.Textbox(label="Query")
39
  k = gr.Slider(1, 50, value=5, label="Number of results")
40
  btn = gr.Button("Search")