model_loading.py

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 get_model_predictions(text):
    headers = {"Authorization": f"Bearer {api_key}"}
    payload = {"inputs": text}
    response = requests.post(f"https://api.huggingface.co/models/{model_path}", headers=headers, json=payload)
    return response.json()