test

app.py

@@ -80,15 +80,24 @@ class MyClient(discord.Client):
             #-> (stage_1_results, stage_1_param_path, stage_1_result_path)
 
             # input prompt
-            prompt = message.content[12:].strip()   
+            prompt = message.content[12:].strip()  
+            
+            negative_prompt = ''
+            seed = 0
             number_of_images = 4
-
-            # random seed
-            current_time = int(time.time())
-            random.seed(current_time)
-            seed = random.randint(0, 2**32 - 1)
+            guidance_scale = 7
+            custom_timesteps_1 = 'smart50'
+            number_of_inference_steps = 50
             
-            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")
+            stage_1_results, stage_1_param_path, stage_1_result_path = client.predict(
+                prompt,
+                negative_prompt,
+                seed,
+                number_of_images,
+                guidance_scale,
+                custom_timesteps_1,
+                number_of_inference_steps,
+                api_name='/generate64')
 
             #stage_1_results, stage_1_param_path, stage_1_result_path = df.predict("gradio written on a wall", "blur", 1,1,7.0, 'smart100',50, api_name="/generate64")
             
@@ -137,10 +146,24 @@ class MyClient(discord.Client):
 
 
 
+            selected_index_for_stage_2 = 0
+            seed_2 = 0
+            guidance_scale_2 = 4
+            custom_timesteps_2 = 'smart50'
+            number_of_inference_steps_2 = 50
+            imgif = client.predict(stage_1_result_path,
+                                    selected_index_for_stage_2,
+                                    seed_2,
+                                    guidance_scale_2,
+                                    custom_timesteps_2,
+                                    number_of_inference_steps_2,
+                                    api_name='/upscale256')
 
-
-
-
+            imgif_path = "/tmp/generated_image.png"
+            imgif.save(imgif_path)
+            
+            with open(imgif_path, 'rb') as f:
+                await message.reply(f'Here is the generated image', file=discord.File(f, 'generated_image.png')) 
 
             '''
             # stage 2