diff --git a/api/views.py b/api/views.py index add133c..9d6fdbe 100644 --- a/api/views.py +++ b/api/views.py @@ -23,7 +23,7 @@ class Comixify(APIView): frames_mode=serializer.validated_data["frames_mode"], rl_mode=serializer.validated_data["rl_mode"], image_assessment_mode=serializer.validated_data["image_assessment_mode"], - style_transfer_mode=serializer.validated_data["rl_mode"], + style_transfer_mode=serializer.validated_data["style_transfer_mode"], ) comic, from_nparray_time = Comic.create_from_nparray(comic_image, video) timings['from_nparray_time'] = from_nparray_time @@ -56,7 +56,7 @@ class ComixifyFromYoutube(APIView): frames_mode=serializer.validated_data["frames_mode"], rl_mode=serializer.validated_data["rl_mode"], image_assessment_mode=serializer.validated_data["image_assessment_mode"], - style_transfer_mode=serializer.validated_data["rl_mode"], + style_transfer_mode=serializer.validated_data["style_transfer_mode"], ) comic, from_nparray_time = Comic.create_from_nparray(comic_image, video) timings['from_nparray_time'] = from_nparray_time