Spaces:
Sleeping
Sleeping
Download app/embedding/embeder.py from Kshitijk20/ClariDoc: direct link, hf CLI and curl.
- Browser
- Download file 336 Bytes
-
https://huggingface.co/spaces/Kshitijk20/ClariDoc/resolve/main/app/embedding/embeder.py
- Command line
-
hf download hf://spaces/Kshitijk20/ClariDoc/app/embedding/embeder.py
-
curl -L -o embeder.py https://huggingface.co/spaces/Kshitijk20/ClariDoc/resolve/main/app/embedding/embeder.py
336 Bytes
| import os | |
| from dotenv import load_dotenv | |
| from typing import Union, List | |
| class QueryEmbedding: | |
| def __init__(self, query: str, embedding_model): | |
| self.query = query | |
| self.embedding_model = embedding_model | |
| def get_embedding(self): | |
| e_main = self.embedding_model.embed_query(self.query) | |
| return e_main | |