推理接口 API:/model/predict 接口

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
您可以使用asyncio库来实现异步推理。以下是一个简单的例子: ```python import asyncio import json import requests async def predict(input_text): # 构建请求数据 data = { "inputs": input_text, "parameters": { "max_response_length": 50, "do_sample": True, "num_return_sequences": 1 } } headers = {"Content-type": "application/json"} # 发送异步请求 response = await loop.run_in_executor(None, requests.post, "https://api-inference.huggingface.co/models/microsoft/DialoGPT-medium/", data=json.dumps(data), headers=headers) # 解析响应 response = json.loads(response.content.decode('utf-8')) return response["generated_text"][0]["text"] async def main(): while True: input_text = input("You: ") if input_text.strip() == "": continue response = await predict(input_text) print("Bot: " + response) if __name__ == "__main__": loop = asyncio.get_event_loop() loop.run_until_complete(main()) ``` 在此示例中,我们定义了一个名为`predict`的异步函数,它接受一个输入文本并返回一个响应文本。在该函数中,我们首先构建了一个包含请求数据的字典,并将其作为JSON字符串发送到Hugging Face API。我们使用`run_in_executor`方法将异步请求转换为同步请求。该方法接受一个可调用对象和一些参数,并在执行器(默认为线程池)中运行该对象。 在`main`函数中,我们使用一个无限循环来等待用户输入,然后调用`predict`函数来获取响应并输出到控制台。在主函数中,我们使用`asyncio.get_event_loop()`来获取事件循环对象,然后使用`run_until_complete`方法来运行主函数。这将启动事件循环并等待直到我们退出程序。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

想做全栈的胖胖

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值