Spaces:
Running
Running
File size: 2,545 Bytes
b78ad07 ec9d18e b78ad07 aba2637 b78ad07 7381c93 b78ad07 7381c93 7ad3a9f c471598 7381c93 b78ad07 7381c93 b78ad07 7ad3a9f aba2637 0af688a 22206cb c471598 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
# convbot
[](https://github.com/ffreemt/convbot/actions)[](https://img.shields.io/static/v1?label=python+&message=3.7%2B&color=blue)[](https://github.com/psf/black)[](https://opensource.org/licenses/MIT)[](https://badge.fury.io/py/convbot)
A conversational bot based on microsoft/DialoGPT-medium
Try it out [](https://colab.research.google.com/github/ffreemt/convbot/blob/master/convbot_demo.ipynb)
## Install it
```shell
pip install convbot
# or poetry add convbot
# pip install git+htts://github.com/ffreemt/convbot
# poetry add git+htts://github.com/ffreemt/convbot
# To upgrade
pip install convbot -U
# or poetry add convbot@latest
```
## Use it
```python
from convbot import convbot
prin(convertbot("How are you?"))
# I am good # or along that line
```
The async version `aconvbot`, potentially for `fastapi` or `Nonebot` plugins and such, is rather artificial since it's based on `ThreadPoolExecutor`. Hence it's not intended for production. You probably should not spawn too many instances.
```python
from convbot import aconvbot
async def afunc(text):
resp = await aconvbot(text)
...
```
Interactive
```bash
python -m convbot
```
2021-07-24 03:11:19.748518: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library libcudart.so.11.0
Bot: Talk to me (type quit to exit)
You: How are you?
Bot: I'm good, you?
You: pretty good. how is the weather there?
Bot: It's pretty cold.
You: really?
Bot: I don't have a heat source.
You:
## Colab testrun `convbot-demo.ipynb`
[](https://colab.research.google.com/github/ffreemt/convbot/blob/master/convbot_demo.ipynb)
## Not tested in Windows 10 and Mac
The module uses pytorch that is installed differently in Windows than in Linux. To run `convbot` in Windows or Mac, you may give it a spin by cloning the repo (git clone [https://github.com/ffreemt/convbot](https://github.com/ffreemt/convbot)) and installing pytorch manually.
|