r2

app.py

@@ -52,49 +52,48 @@ async def deepfloydif(ctx, *, prompt: str):
         def check_reaction(reaction, user):
             return str(reaction.emoji) == '👍' and user == ctx.author 
         
-        with rate_limiter:
-            number_of_images = 4
-            current_time = int(time.time())
-            random.seed(current_time)
-            seed = random.randint(0, 2**32 - 1)
-            stage_1_results, stage_1_param_path, stage_1_result_path = df.predict(prompt, "blur", seed, number_of_images, 7.0, 'smart100', 50, api_name="/generate64")
-            png_files = [f for f in os.listdir(stage_1_results) if f.endswith('.png')]
-
-            if png_files:
-                first_png = png_files[0]
-                second_png = png_files[1]
-                third_png = png_files[2]
-                fourth_png = png_files[3]
-
-                first_png_path = os.path.join(stage_1_results, first_png)
-                second_png_path = os.path.join(stage_1_results, second_png)
-                third_png_path = os.path.join(stage_1_results, third_png)
-                fourth_png_path = os.path.join(stage_1_results, fourth_png)
-
-                img1 = Image.open(first_png_path)
-                img2 = Image.open(second_png_path)
-                img3 = Image.open(third_png_path)
-                img4 = Image.open(fourth_png_path)
-
-                combined_image = Image.new('RGB', (img1.width * 2, img1.height * 2))
-
-                combined_image.paste(img1, (0, 0))
-                combined_image.paste(img2, (img1.width, 0))
-                combined_image.paste(img3, (0, img1.height))
-                combined_image.paste(img4, (img1.width, img1.height))
-
-                combined_image_path = os.path.join(stage_1_results, 'combined_image.png')
-                combined_image.save(combined_image_path)
-
-                # Trigger the second stage prediction
-                #await dfif2(ctx, stage_1_result_path)
-
-            await ctx.reply('Here is the combined image. Select an option quickly!')
-            with open(combined_image_path, 'rb') as f:
-                await ctx.send(file=discord.File(f, 'combined_image.png')) 
-
-            reaction, user = await bot.wait_for('reaction_add', check=check_reaction)
-            await ctx.send(f"{user.mention} reacted with a thumbs-up!")                
+        number_of_images = 4
+        current_time = int(time.time())
+        random.seed(current_time)
+        seed = random.randint(0, 2**32 - 1)
+        stage_1_results, stage_1_param_path, stage_1_result_path = df.predict(prompt, "blur", seed, number_of_images, 7.0, 'smart100', 50, api_name="/generate64")
+        png_files = [f for f in os.listdir(stage_1_results) if f.endswith('.png')]
+
+        if png_files:
+            first_png = png_files[0]
+            second_png = png_files[1]
+            third_png = png_files[2]
+            fourth_png = png_files[3]
+
+            first_png_path = os.path.join(stage_1_results, first_png)
+            second_png_path = os.path.join(stage_1_results, second_png)
+            third_png_path = os.path.join(stage_1_results, third_png)
+            fourth_png_path = os.path.join(stage_1_results, fourth_png)
+
+            img1 = Image.open(first_png_path)
+            img2 = Image.open(second_png_path)
+            img3 = Image.open(third_png_path)
+            img4 = Image.open(fourth_png_path)
+
+            combined_image = Image.new('RGB', (img1.width * 2, img1.height * 2))
+
+            combined_image.paste(img1, (0, 0))
+            combined_image.paste(img2, (img1.width, 0))
+            combined_image.paste(img3, (0, img1.height))
+            combined_image.paste(img4, (img1.width, img1.height))
+
+            combined_image_path = os.path.join(stage_1_results, 'combined_image.png')
+            combined_image.save(combined_image_path)
+
+            # Trigger the second stage prediction
+            #await dfif2(ctx, stage_1_result_path)
+
+        await ctx.reply('Here is the combined image. Select an option quickly!')
+        with open(combined_image_path, 'rb') as f:
+            await ctx.send(file=discord.File(f, 'combined_image.png')) 
+
+        reaction, user = await bot.wait_for('reaction_add', check=check_reaction)
+        await ctx.send(f"{user.mention} reacted with a thumbs-up!")                
 
     except Exception as e:
         print(f"Error: {e}")