test

test.py

@@ -0,0 +1,32 @@
+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)