如何使用 API 与 Google Colab 交互？

Question

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

Answer 1

使用 python 和 api 的方法之一：在此处创建配置文件：https://huggingface.co/settings/profile 您可能需要将 API_URL 替换为您自己的。我相信您可以使用邮递员创建一个 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.",
        }
    }
)