import os
from dotenv import load_dotenv
from sentence_transformers import SentenceTransformer
load_dotenv()

API_TOKEN = os.getenv('HF_TOKEN')

API_URL = "/static-proxy?url=https%3A%2F%2Fapi-inference.huggingface.co%2Fmodels%2Fall-distilroberta-v1"
headers = {"Authorization": f"Bearer {API_TOKEN}"}


def generate_text_embeddings(texts):
    model = SentenceTransformer('all-distilroberta-v1')

    return model.encode(texts)

# Example usage: