File size: 343 Bytes
b692cf4
397fd15
b692cf4
 
 
397fd15
8bad412
b692cf4
8bad412
397fd15
 
 
1
2
3
4
5
6
7
8
9
10
11
12
import os
import requests
from dotenv import load_dotenv

load_dotenv()
api_key = os.getenv('HF_API_KEY')
model_path = os.getenv('MODEL_PATH')

def load_model():
    # For Hugging Face, loading a model is typically done via an API request rather than locally.
    headers = {"Authorization": f"Bearer {api_key}"}
    return headers, model_path