CodeMirror/Brandon209-Red-bot-Cogs
1
0
Fork 0
mirror of https://github.com/brandons209/Red-bot-Cogs.git synced 2024-06-28 19:10:34 +12:00
Code Issues Projects Releases Wiki Activity

fix async calls

Browse source
This commit is contained in:
brandons209 2019-01-20 02:37:53 -05:00
parent 75d7e1a40d
commit 89760ed222
1 changed files with 7 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
scriptcog/script.py
View file

@ -148,7 +148,7 @@ class ScriptCog:
int_text = np.array([self.word_to_int[word] for word in int_text], dtype=np.int32) int_text = np.array([self.word_to_int[word] for word in int_text], dtype=np.int32)
except KeyError: except KeyError:
await self.bot.say("Sorry, that seed word is not in my vocabulary.\nPlease try an English word from the show.\n") await self.bot.say("Sorry, that seed word is not in my vocabulary.\nPlease try an English word from the show.\n")
return None return
#pad text if it is too short, pads with zeros at beginning of text, so shouldnt have too much noise added #pad text if it is too short, pads with zeros at beginning of text, so shouldnt have too much noise added
int_text = pad_sequences([int_text], maxlen=self.sequence_length) int_text = pad_sequences([int_text], maxlen=self.sequence_length)
#predict next word: #predict next word:
@ -157,7 +157,10 @@ class ScriptCog:
#append to the result #append to the result
input_text += ' ' + output_word input_text += ' ' + output_word
#convert tokenized punctuation and other characters back #convert tokenized punctuation and other characters back
return _untokenize_punctuation(input_text) result = _untokenize_punctuation(input_text)
await self.bot.say("------------------------")
await self.bot.say(result)
await self.bot.say("------------------------")
@commands.command(pass_context=True, no_pm=True) @commands.command(pass_context=True, no_pm=True)
async def genscript(self, ctx, num_words_to_generate : int = 100, variance : float = 0.5, seed : str = "pinkie pie::"): async def genscript(self, ctx, num_words_to_generate : int = 100, variance : float = 0.5, seed : str = "pinkie pie::"):
@ -179,13 +182,8 @@ class ScriptCog:
variance = 0 variance = 0
await self.bot.say("Generating script, please wait...") await self.bot.say("Generating script, please wait...")
loop = asyncio.get_event_loop()
result = await self.get_model_output(num_words_to_generate, variance, seed) await self.get_model_output(num_words_to_generate, variance, seed)
if result is not None:
await self.bot.say("------------------------")
await self.bot.say(result)
await self.bot.say("------------------------")
def setup(bot): def setup(bot):
bot.add_cog(ScriptCog(bot)) bot.add_cog(ScriptCog(bot))