StreamlitChat / utils.py
pragneshbarik's picture
initial commit
d60dac5
raw
history blame
No virus
445 Bytes
import json
import requests
import os
from dotenv import load_dotenv
from sentence_transformers import SentenceTransformer
load_dotenv()
API_TOKEN = os.getenv('HF_TOKEN')
API_URL = "https://api-inference.huggingface.co/models/all-distilroberta-v1"
headers = {"Authorization": f"Bearer {API_TOKEN}"}
def generate_text_embeddings(texts):
model = SentenceTransformer('all-distilroberta-v1')
return model.encode(texts)
# Example usage: