lunarflu HF Staff commited on
Commit
a598b3b
·
1 Parent(s): 8dfe030
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -51,7 +51,7 @@ class ButtonView(discord.ui.View):
51
  self.stop()
52
 
53
  async def button_handler(self, index: int):
54
- await self.ctx.invoke(self.ctx.bot.get_command('dfif2'), index=index, stage_1_result_path=self.stage_1_result_path)
55
  self.stop()
56
 
57
  @discord.ui.button(label='Image 1', style=discord.ButtonStyle.blurple)
@@ -369,7 +369,7 @@ async def deepfloydif(ctx, *, prompt: str):
369
  @commands.cooldown(1, 5, commands.BucketType.user)
370
  async def dfif2(ctx, index: int, stage_1_result_path):
371
  try:
372
- image_path = self.image_paths[index]
373
 
374
  selected_index_for_stage_2 = image_path
375
  seed_2 = 0
 
51
  self.stop()
52
 
53
  async def button_handler(self, index: int):
54
+ await self.ctx.invoke(self.ctx.bot.get_command('dfif2'), index=index, stage_1_result_path=self.stage_1_result_path, image_paths=self.image_paths)
55
  self.stop()
56
 
57
  @discord.ui.button(label='Image 1', style=discord.ButtonStyle.blurple)
 
369
  @commands.cooldown(1, 5, commands.BucketType.user)
370
  async def dfif2(ctx, index: int, stage_1_result_path):
371
  try:
372
+ image_path = image_paths[index]
373
 
374
  selected_index_for_stage_2 = image_path
375
  seed_2 = 0