Spaces:

LordFarquaad42
/

Groove-GPT

Running

LordFarquaad42 commited on Apr 29

Commit

c796de9

•

1 Parent(s): 1612f56

fixed bug where create_client loaded client

Files changed (1) hide show

add_data.py CHANGED Viewed

@@ -9,7 +9,7 @@ def create_client():
     EMBEDDING_FUNC = embedding_functions.SentenceTransformerEmbeddingFunction(
         model_name=MODEL_NAME
     )
-    schemer = client.get_collection(
         name=COLLECTION_NAME,
         embedding_function=EMBEDDING_FUNC,
     )

     EMBEDDING_FUNC = embedding_functions.SentenceTransformerEmbeddingFunction(
         model_name=MODEL_NAME
     )
+    schemer = client.create_collection(
         name=COLLECTION_NAME,
         embedding_function=EMBEDDING_FUNC,
     )