lunarflu HF Staff commited on
Commit
0e6b97a
·
1 Parent(s): 7fa9e07

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -46,9 +46,10 @@ async def discord_before(ctx):
46
  return False
47
 
48
  async def generation(attachment):
49
- style = 'JoJo'
50
- im = jojogan.predict(attachment.url, style)
51
- return im
 
52
 
53
  async def discord_after(ctx, im):
54
  await ctx.send(f'{ctx.author.mention} Here is the jojo version of it')
 
46
  return False
47
 
48
  async def generation(attachment):
49
+ if attachment not False:
50
+ style = 'JoJo'
51
+ im = jojogan.predict(attachment.url, style)
52
+ return im
53
 
54
  async def discord_after(ctx, im):
55
  await ctx.send(f'{ctx.author.mention} Here is the jojo version of it')