Added Wolfram!

This commit is contained in:
Robert 2020-02-20 22:52:00 +01:00
parent 5c4c6631ba
commit 526d9d72ef
4 changed files with 27 additions and 38 deletions

View file

@ -1,17 +1,14 @@
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
@ -19,12 +16,6 @@ class Responses(commands.Cog):
await message.channel.send(f"Stop pinging us {message.author.mention} <:pinged:451198700832817202>")
return
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:
if message.content == "<@&380535423233032193> NUMBERS REEEEEEEE":
await message.channel.send(f"<@&380535423233032193> {message.author.mention} REEEEEEEE")