Update app.py

app.py

@@ -427,7 +427,7 @@ async def dfif2(index: int, stage_1_result_path, thread, dfif_command_message_id
         parent_channel = thread.parent
         dfif_command_message = await parent_channel.fetch_message(dfif_command_message_id)
         await dfif_command_message.add_reaction(success_emoji)
-        await ctx.message.remove_reaction('⏳')
+        await dfif_command_message.remove_reaction('⏳')
 
     except Exception as e:
         print(f"Error: {e}")
@@ -438,7 +438,7 @@ async def dfif2(index: int, stage_1_result_path, thread, dfif_command_message_id
         parent_channel = thread.parent
         dfif_command_message = await parent_channel.fetch_message(dfif_command_message_id)
         await dfif_command_message.add_reaction(failure_emoji)  
-        await ctx.message.remove_reaction('⏳')
+        await dfif_command_message.remove_reaction('⏳')
         await thread.send(f"Error during stage 2 upscaling.") 
 #----------------------------------------------------------------------------------------------------------------------------
 # react detector for stage 2 ✅