Update app.py

app.py

@@ -3,25 +3,13 @@ from transformers import pipeline
 import torch
 from diffusers import StableDiffusionPipeline
 
-def get_completion(prompt):
-    return pipeline(prompt)['sample'][0]
-    
-def generate(prompt, negative_prompt, steps, guidance, width, height):
-    params = {
-        "negative_prompt": negative_prompt,
-        "num_inference_steps": steps,
-        "guidance_scale": guidance,
-        "width": width,
-        "height": height
-    }
+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]
 
-    output = get_completion(prompt)
-    # pil_image = base64_to_pil(output)
-    # return pil_image
+def generate(prompt,negative_prompt,steps,guidance,width,height):
+    output = get_completion(prompt=prompt, height=height, width=width, steps=steps, guidance=guidance, negative_prompt=negative_prompt)
     return output
 
-pipeline = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
-
 with gr.Blocks() as demo:
     gr.Markdown("# Image Generation with Stable Diffusion")
     with gr.Row():