hugginggpt/huggingface_api.py

import os

from dotenv import load_dotenv

load_dotenv()

HUGGINGFACE_INFERENCE_API_URL = "https://api-inference.huggingface.co/models/"
HUGGINGFACE_INFERENCE_API_STATUS_URL = f"https://api-inference.huggingface.co/status/"


def get_hf_headers():
    HUGGINGFACEHUB_API_TOKEN = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
    return {"Authorization": f"Bearer {HUGGINGFACEHUB_API_TOKEN}"}