Spaces:
Running
Running
Update app.py
Browse files
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 = "
|
11 |
-
embedding_column = "
|
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 [
|
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")
|