diff --git a/cogs/fun/responses.py b/cogs/fun/responses.py index 41ba0fd..a2dbcd1 100644 --- a/cogs/fun/responses.py +++ b/cogs/fun/responses.py @@ -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: diff --git a/data/textgenrnn_weights.hdf5 b/data/textgenrnn_weights.hdf5 new file mode 100644 index 0000000..bbecbf9 Binary files /dev/null and b/data/textgenrnn_weights.hdf5 differ