Update app.py

app.py

@@ -7,6 +7,7 @@ from huggingface_hub import get_token
 
 static_embedding = StaticEmbedding.from_model2vec("minishlab/potion-base-8M")
 model = SentenceTransformer(modules=[static_embedding])
+embedding_dimensions = model.get_sentence_embedding_dimension()
 dataset_name = "ai-blueprint/fineweb-bbc-news-embeddings"
 embedding_column = "embeddings"
 table_name = "fineweb"
@@ -15,7 +16,7 @@ duckdb.sql(query=f"""
     INSTALL vss;
     LOAD vss;
     CREATE TABLE {table_name} AS 
-    SELECT *, {embedding_column}::float[{model.get_sentence_embedding_dimension()}] as embedding_float 
+    SELECT *, {embedding_column}::float[{embedding_dimensions}] as {embedding_column}_float 
     FROM 'hf://datasets/{dataset_name}/**/*.parquet';
     CREATE INDEX my_hnsw_index ON {table_name} USING HNSW (embedding_float) WITH (metric = 'cosine');
 """)
@@ -24,7 +25,7 @@ def similarity_search(query: str, k: int = 5):
     embedding = model.encode(query).tolist()
     return duckdb.sql(
         query=f"""
-        SELECT chunk, url, array_cosine_distance({embedding_column}_float, {embedding}::FLOAT[{model.get_sentence_embedding_dimension()}]) as distance 
+        SELECT chunk, url, array_cosine_distance({embedding_column}_float, {embedding}::FLOAT[{embedding_dimensions}]) as distance 
         FROM {table_name}
         ORDER BY distance 
         LIMIT {k};
@@ -34,7 +35,7 @@ def similarity_search(query: str, k: int = 5):
 with gr.Blocks() as demo:
     gr.Markdown("""# RAG - retrieve
                 
-                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. """)
+                Part of [AI blueprint](https://github.com/huggingface/ai-blueprint) - a blueprint for AI development, focusing on practical examples of RAG, information extraction, analysis and fine-tuning in the age of LLMs. """)
     query = gr.Textbox(label="Query")
     k = gr.Slider(1, 50, value=5, label="Number of results")
     btn = gr.Button("Search")