Update app.py

app.py

@@ -122,7 +122,7 @@ def end_session(req: gr.Request):
     user_dir = os.path.join(TMP_DIR, str(req.session_hash))
     shutil.rmtree(user_dir)
 
-
+@spaces.GPU
 def preprocess_image(image: Image.Image, 
                     prompt: str = "",
                     negative_prompt: str = "",
@@ -557,8 +557,8 @@ if __name__ == "__main__":
     pipe_control.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe_control.scheduler.config)
     pipe_control.to(device)
     
-    # try:
-    #     pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8)))    # Preload rembg
-    # except:
-    #     pass
+    try:
+        pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8)))    # Preload rembg
+    except:
+        pass
     demo.launch()