Update app.py

app.py

@@ -93,6 +93,7 @@ def resize_image_with_aspect(image, res_range_min=128, res_range_max=1024):
 
 @spaces.GPU()
 def reconstruct(input_img, caption):
+    pipe, inverse_scheduler, scheduler = load_pipeline()
     global weights
     weights = {}
 
@@ -478,8 +479,11 @@ def load_pipeline():
         load_file(hf_hub_download(repo_id="jiaxiangc/res-adapter", subfolder=resadapter_model_name, filename="diffusion_pytorch_model.safetensors")),
         strict=False,
         ) # load norm weights
+
+    inverse_scheduler = DDIMInverseScheduler.from_pretrained(model_id, subfolder="scheduler")
+    scheduler = DDIMScheduler.from_pretrained(model_id, subfolder="scheduler")
     
-    return pipe
+    return pipe, inverse_scheduler, scheduler
     
 if __name__ == "__main__":
 
@@ -507,12 +511,6 @@ if __name__ == "__main__":
 
     torch_dtype = torch.float16
 
-    # torch_dtype = torch.float16
-    pipe = load_pipeline()
-
-    inverse_scheduler = DDIMInverseScheduler.from_pretrained(model_id, subfolder="scheduler")
-    scheduler = DDIMScheduler.from_pretrained(model_id, subfolder="scheduler")
-
 
 with gr.Blocks(analytics_enabled=False) as demo:
     gr.Markdown(