File size: 936 Bytes
ea55fae
3cac10f
ea55fae
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import gradio as gr
import os


HF_TOKEN = os.getenv('HF_TOKEN')
hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "Rick-bot-flags")

title = "Have Fun With RickBot"
description = """
<p>
<center>
The bot is trained on Rick and Morty dialogues Kaggle Dataset using DialoGPT.
<img src="https://gradio.app/assets/img/rick.gif">
</center>
</p>
"""
article = "<p style='text-align: center'><a href='https://medium.com/geekculture/discord-bot-using-dailogpt-and-huggingface-api-c71983422701' target='_blank'>Complete Tutorial</a></p><p style='text-align: center'><a href='https://dagshub.com/kingabzpro/DailoGPT-RickBot' target='_blank'>Project is Available at DAGsHub</a></p></center></p>"

gr.Interface.load("kingabzpro/DialoGPT-small-Rick-Bot", 
    "textbox","chatbot",theme ="grass", title = title, flagging_callback=hf_writer,description = description, article = article).launch(enable_queue=True)  # customizes the input component