Update app.py

app.py

@@ -3,11 +3,19 @@ from transformers import pipeline
 import torch
 from diffusers import StableDiffusionPipeline
 
-def get_completion(prompt,negative_prompt,steps,guidance,width,height):
-    return pipeline(prompt=prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=guidance, negative_prompt=negative_prompt)['sample'][0]
+def get_completion(prompt,params):
+    # return pipeline(prompt=prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=guidance, negative_prompt=negative_prompt)['sample'][0]
+    return pipeline(prompt=prompt, params=params)['sample'][0]
 
 def generate(prompt,negative_prompt,steps,guidance,width,height):
-    output = get_completion(prompt,negative_prompt,steps,guidance,width,height)
+    params = {
+        "negative_prompt": negative_prompt,
+        "num_inference_steps": steps,
+        "guidance_scale": guidance,
+        "width": width,
+        "height": height
+    }
+    output = get_completion(prompt,params)
     return output
     
 pipeline = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)