Upload app.py

app.py

@@ -175,7 +175,7 @@ def contrast_loss(images):
     variance = torch.var(images)
     return -variance
 
-def generate_with_prompt_style_guidance(prompt, style, seed):
+def generate_with_prompt_style_guidance(prompt, style, seed,num_inference_steps,guidance_scale):
 
     prompt = prompt + ' in style of s'
     
@@ -183,8 +183,8 @@ def generate_with_prompt_style_guidance(prompt, style, seed):
 
     height = 512                        # default height of Stable Diffusion
     width = 512                         # default width of Stable Diffusion
-    num_inference_steps = 10  #           # Number of denoising steps
-    guidance_scale = 8 #               # Scale for classifier-free guidance
+    num_inference_steps = num_inference_steps  #           # Number of denoising steps
+    guidance_scale = guidance_scale #               # Scale for classifier-free guidance
     generator = torch.manual_seed(seed)   # Seed generator to create the inital latent noise
     batch_size = 1
     contrast_loss_scale = 200 #