Skip to content

mihirdeo16/ml_text_gen_chatbot

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ML Chatbot

Text_genration bot using GTP-2 model (microsoft/DialoGPT-medium) You can deplot the bot using Fastapi by running command :

pip install requirements.txt
cd src/
uvicorn chatbot:app --reload

Also you can check the output in termial using Curl command line utitlity

curl -X GET "http://127.0.0.1:8000/chat?text=Hello%2C%20how%20are%20you%3F" \
 -H "accept: application/json"
 
 >>> { "Answer":"I'm good, how are you?"}

Example : Terminal/output

Alternative way is you Fastapi GUI service which can be access by:

http://127.0.0.1:8000/docs

About

Building chat_bot from transformers

Topics

Resources

Stars

Watchers

Forks

Languages