Update app.py

app.py

@@ -76,7 +76,8 @@ class DemoUI(object):
         pipe_cfg = self.model_choices[self.default_model_name]
         infer_name = pipe_cfg.get("INFERENCE_TYPE", "ACE")
         self.pipe = inference_dict[infer_name]()
-        self.pipe.init_from_cfg(pipe_cfg)
+        with spaces.GPU(duration=60):
+            self.pipe.init_from_cfg(pipe_cfg)
 
         # choose different model
         self.task_model_cfg = Config(load=True, cfg_file=model_list)