Update deepfloydif.py

deepfloydif.py

@@ -24,7 +24,7 @@ def deepfloydif_stage_1_inference(prompt):
     guidance_scale = 7
     custom_timesteps_1 = 'smart50'
     number_of_inference_steps = 50
-    stage_1_results, stage_1_param_path, stage_1_result_path = deepfloyd_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 = deepfloydif_client.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]    
 
 def deepfloydif_stage_2_inference(index, stage_1_result_path):
@@ -34,7 +34,7 @@ def deepfloydif_stage_2_inference(index, stage_1_result_path):
     guidance_scale_2 = 4
     custom_timesteps_2 = 'smart50'
     number_of_inference_steps_2 = 50
-    result_path = deepfloyd_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')
+    result_path = deepfloydif_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')
     return result_path