Update app.py

app.py

@@ -291,6 +291,21 @@ async def react1234(reaction_emojis, combined_image_dfif):
 
 
 #----------------------------------------------------------------------------------------------------------------------------
+#----------------------------------------------------------------------------------------------------------------------------------------------  
+# deepfloydif stage 1 generation ✅
+def inference(prompt):
+    negative_prompt = ''
+    seed = random.randint(0, 1000)
+    #seed = 1
+    number_of_images = 4
+    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, negative_prompt, seed, number_of_images, guidance_scale, custom_timesteps_1, number_of_inference_steps, api_name='/generate64')
+    
+    return [stage_1_results, stage_1_param_path, stage_1_result_path]        
 #----------------------------------------------------------------------------------------------------------------------------        
 # running in thread
 DISCORD_TOKEN = os.environ.get("DISCORD_TOKEN", None)