Update app.py

app.py

@@ -9,6 +9,8 @@ def get_completion(prompt,negative_prompt,steps,guidance,width,height):
 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")