main

app.py

@@ -144,24 +144,7 @@ class MyClient(discord.Client):
             with open(combined_image_path, 'rb') as f:
                 await message.reply('Here is the combined image', file=discord.File(f, 'combined_image.png'))
 
-            '''
-            # stage 2
-            selected_index_for_stage_2 = -1
-            custom_timesteps_2 = 'smart100' # could reset to smart50 if index was the issue
-            seed = 362572064 # again, could randomize this        seed = 362572064      seed = random.randint(0, 2**32 - 1)
-            
-            # predict(stage_1_result_path, selected_index_for_stage_2, seed, guidance_scale, custom_timesteps_2, number_of_inference_steps, api_name="/upscale256") -> result
-            
-            img = df.predict(stage_1_result_path, selected_index_for_stage_2, seed, guidance_scale, custom_timesteps_2, number_of_inference_steps, api_name="/upscale256")
-    
-            # Save the generated image to a file
-            img_path = "/tmp/generated_image.png"
-            img.save(img_path)
-            
-            # Send the image file as a Discord attachment
-            with open(img_path, 'rb') as f:
-                await message.reply(f'Here is the generated image', file=discord.File(f, 'generated_image.png'))                
-            '''
+