diff --git a/api/models.py b/api/models.py
index ea5e219..8a836b8 100644
--- a/api/models.py
+++ b/api/models.py
@@ -35,9 +35,11 @@ class Video(models.Model):
             raise TooLargeFile()
 
     def create_comic(self, frames_mode=0, rl_mode=0):
-        keyframes, keyframes_extraction_time = KeyFramesExtractor.get_keyframes(video=self,
-                                                                                frames_mode=frames_mode,
-                                                                                rl_mode=rl_mode)
+        keyframes, keyframes_extraction_time = KeyFramesExtractor.get_keyframes(
+            video=self,
+            frames_mode=frames_mode,
+            rl_mode=rl_mode
+        )
         stylized_keyframes, stylization_time = StyleTransfer.get_stylized_frames(frames=keyframes)
         comic_image, layout_generation_time = LayoutGenerator.get_layout(frames=stylized_keyframes)
 
diff --git a/keyframes/keyframes.py b/keyframes/keyframes.py
index caf8635..7918a44 100644
--- a/keyframes/keyframes.py
+++ b/keyframes/keyframes.py
@@ -1,6 +1,6 @@
 import os
 import uuid
-
+import shutil
 import numpy as np
 import torch
 import torch.nn as nn
@@ -37,6 +37,7 @@ class KeyFramesExtractor:
         probs = cls._get_probs(norm_features, gpu, mode=rl_mode)
         keyframes = cls._get_keyframes(frames, probs, change_points, frames_per_segment)
         chosen_frames = cls._get_popularity_chosen_frames(keyframes, features)
+        shutil.rmtree(jj(f"{settings.TMP_DIR}", f"{all_frames_tmp_dir}"))
         return chosen_frames
 
     @staticmethod