import sys
import os
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..')))
from huggingface_hub import InferenceClient
from backend.app.config import config

def verify():
    print("Verifying token with gpt2...")
    try:
        client = InferenceClient(token=config.HUGGINGFACE_API_KEY)
        res = client.text_generation("Hello", model="gpt2")
        print("Success with GPT2!")
        print(res)
    except Exception as e:
        print(f"Failed with GPT2: {e}")

if __name__ == "__main__":
    verify()