Spaces:

Dinesh310
/

Demo_1

Sleeping

Dinesh310 commited on Jan 25

Commit

3305f51

verified ·

1 Parent(s): efca102

Update src/RAG_builder.py

Files changed (1) hide show

src/RAG_builder.py CHANGED Viewed

@@ -19,7 +19,7 @@ class GraphState(TypedDict):
 class ProjectRAGGraph:
     def __init__(self):
         self.embeddings = HuggingFaceEmbeddings(
-            model_name="sentence-transformers/all-MiniLM-L6-v2",
             model_kwargs={"device": "cpu"},
             encode_kwargs={"normalize_embeddings": True}
         )

 class ProjectRAGGraph:
     def __init__(self):
         self.embeddings = HuggingFaceEmbeddings(
+            model_name="google/embeddinggemma-300m",
             model_kwargs={"device": "cpu"},
             encode_kwargs={"normalize_embeddings": True}
         )