Update app.py

app.py

@@ -249,7 +249,7 @@ def infer(prompt, samples, steps, scale, seed, global_info):
     
             big_video_list = []
     
-            global_info['decode_res'] = decode_res
+            global_info['decode_res'] = decode_res.detach()
     
             for b in range(batch_size):
                 def render_img(v):
@@ -303,6 +303,7 @@ def infer(prompt, samples, steps, scale, seed, global_info):
     
             path = f"tmp/{prompt.replace(' ', '_')}_{str(datetime.datetime.now()).replace(' ', '_')}.mp4"
             imageio.mimwrite(path, np.stack(cat_video_list, 0))
+            torch.cuda.empty_cache()
 
     return global_info, path