Update app.py

app.py

@@ -40,11 +40,13 @@ class Pipeline:
         self.denoiser = denoiser.cuda()
         self.conditioner = conditioner.cuda()
         self.resolution = resolution
-        self.tmp_dir = tempfile.TemporaryDirectory(prefix="traj_gifs_")
+        self.tmp_dir = "./traj_gifs_"
+
+        # 若文件夹已存在，先删除旧文件（可选，根据需求决定）
+        if os.path.exists(self.tmp_dir):
+            shutil.rmtree(self.tmp_dir)  # 递归删除文件夹及内容
         # self.denoiser.compile()
 
-    def __del__(self):
-        self.tmp_dir.cleanup()
 
     @spaces.GPU(duration=5)
     @torch.no_grad()