python - 如何使用 API 与 Google Colab 交互？

Question

是否可以使用 API 与 Google Colab 环境进行交互？我想根据一些用户通过 API 的输入在 GPU 上编写和执行 python 代码，并将响应返回给用户。我可以使用 Colab 执行此操作吗？或者我是否需要设置自己的 Jupyter Notebook 来执行此操作。

Answer 1

有一种使用 python 和 api 的方法:在此处创建配置文件:https://huggingface.co/settings/profile 您可能必须将 API_URL 替换为您自己的。我相信您可以使用 postman 创建 API_URL，但我不是 100% 确定。

import json
import requests
API_URL = "https://api-inference.huggingface.co/models/deepset/roberta-base-squad2"
headers = {"Authorization": f"Bearer {API_TOKEN}"}

def query(payload):
    data = json.dumps(payload)
    response = requests.request("POST", API_URL, headers=headers, data=data)
    return json.loads(response.content.decode("utf-8"))
data = query(
    {
        "inputs": {
            "question": "What is the purpose of Spark",
            "context": "The purpose of Spark is to help scientists find and better understand the potential causes of autism.",
        }
    }
)