Spaces:

davanstrien
/

huggingface-datasets-search-v2

Running on CPU Upgrade

davanstrien HF Staff commited on Sep 18, 2024

Commit

1d40966

1 Parent(s): 6a4b44c

update model

Files changed (1) hide show

main.py CHANGED Viewed

@@ -212,7 +212,7 @@ async def api_query_by_text(query: str, n: int = Query(default=10, ge=1, le=100)
 async def api_search_viewer(query: str, n: int = Query(default=10, ge=1, le=100)):
     try:
         embedding_function = SentenceTransformerEmbeddingFunction(
-            model_name="davanstrien/dataset-viewer-descriptions-processed-st",
             trust_remote_code=True,
         )
         collection = client.get_collection(

 async def api_search_viewer(query: str, n: int = Query(default=10, ge=1, le=100)):
     try:
         embedding_function = SentenceTransformerEmbeddingFunction(
+            model_name="davanstrien/query-to-dataset-viewer-descriptions",
             trust_remote_code=True,
         )
         collection = client.get_collection(