added rnn

This commit is contained in:
Robert 2020-02-05 17:04:49 +01:00
parent 03fc4cadff
commit 5c4c6631ba
2 changed files with 10 additions and 2 deletions

View file

@ -1,20 +1,28 @@
import discord
from discord.ext import commands
from textgenrnn import textgenrnn
class Responses(commands.Cog):
def __init__(self, client):
self.client = client
self.model = textgenrnn("data/textgenrnn_weights.hdf5")
@commands.Cog.listener()
async def on_message(self, message: discord.Message):
if message.author == self.client.user:
print("Cock")
return
# @applesauce
if ("<@!568271450176356352>" in message.content) and ("<@!657709911337074698>" in message.content):
await message.channel.send(f"Stop pinging us {message.author.mention} <:pinged:451198700832817202>")
return
if "<@!657709911337074698>" in message.content:
await message.channel.send(f"{message.author.mention} <:pinged:451198700832817202>")
if "<@!657709911337074698>" in message.content:
print("In")
result = self.model.generate(temperature=0.7, return_as_list=True)[0]
await message.channel.send(result)
return
if message.author.id == 478006431589728259:

Binary file not shown.