Spaces:

Nevidu
/

IIT_ChatBot

Sleeping

Nevidu commited on Feb 14

Commit

8cc3609

verified ·

1 Parent(s): 2329e0a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -38,7 +38,7 @@ generator = pipeline(
 vectorstore = Chroma(
     # docs,
-    embedding_function=HuggingFaceEmbeddings(model_name = "google/gemma-2b"),
     persist_directory="chroma_db"
 )

 vectorstore = Chroma(
     # docs,
+    embedding_function=HuggingFaceEmbeddings(model_name = "sentence-transformers/all-MiniLM-L6-v1"),
     persist_directory="chroma_db"
 )