Hugging Face's logo

Spaces:
mikeee
/
convbot
Runtime error

App Files Community
1
convbot
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
[![tests](https://github.com/ffreemt/convbot/actions/workflows/routine-tests.yml/badge.svg)](https://github.com/ffreemt/convbot/actions)[![python](https://img.shields.io/static/v1?label=python+&message=3.7%2B&color=blue)](https://img.shields.io/static/v1?label=python+&message=3.7%2B&color=blue)[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)[![PyPI version](https://badge.fury.io/py/convbot.svg)](https://badge.fury.io/py/convbot)

A conversational bot based on microsoft/DialoGPT-medium

Try it out [![Open convbot-demo.ipynb In Colab](https://colab.research.google.com/assets/colab-badge.svg)](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`

[![Open convbot-demo.ipynb In Colab](https://colab.research.google.com/assets/colab-badge.svg)](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.