pipe definition

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
 import torch
 
+pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, revision="fp16", torch_dtype=torch.float16)
+
 def txt2img(prompt):
   image = pipe(prompt, height=768, width=768, guidance_scale = 10).images[0]
   # image.save("sd_image.png")