import os
import requests

# 从环境变量中读取访问令牌
access_token = os.getenv("HF_TOKEN")  # HF_TOKEN 是你在 Secrets 中设置的名称

# API 地址
API_URL = "https://api-inference.huggingface.co/models/gpt2"

# 输入数据
input_data = {
    "inputs": "Once upon a time",
    "parameters": {
        "max_length": 50,  # 生成文本的最大长度
        "temperature": 0.7  # 控制生成文本的随机性
    }
}

# 添加访问令牌到请求头中
headers = {
    "Authorization": f"Bearer {access_token}"  # 使用环境变量中的访问令牌
}

# 发送请求
response = requests.post(API_URL, headers=headers, json=input_data)

# 解析结果
if response.status_code == 200:
    generated_text = response.json()[0]['generated_text']
    print("Generated Text:", generated_text)
else:
    print("Error:", response.status_code, response.text)