update

model.py

@@ -12,10 +12,10 @@ device = "cpu"
 class Model:
     def __init__(self):
         modelID = "runwayml/stable-diffusion-v1-5"
-        pipe = StableDiffusionPipeline.from_pretrained(modelID, torch_dtype=torch.float16)
-        pipe = pipe.to(device)
-        pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
-        
+        self.pipe = StableDiffusionPipeline.from_pretrained(modelID, torch_dtype=torch.float16)
+        self.pipe = self.pipe.to(device)
+        self.pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
+
         #self.pipe = StableDiffusionPipeline.from_pretrained(modelID)
         #prompt = "a photo of an astronaut riding a horse on mars"
         #n_prompt = "deformed, disfigured"