Update app.py

app.py

@@ -1,6 +1,8 @@
 import os
 import cv2
+import torch
 import base64
+import argparse
 import numpy as np
 import insightface
 import onnxruntime
@@ -71,7 +73,7 @@ def load_face_swapper_model(path="./assets/pretrained_models/inswapper_128.onnx"
 def load_face_parser_model(path="./assets/pretrained_models/79999_iter.pth"):
     global FACE_PARSER
     if FACE_PARSER is None:
-        FACE_PARSER = init_parsing_model(path, device='cpu')
+        FACE_PARSER = init_parsing_model(path, device=device)
 
 # Загрузка всех моделей
 load_face_analyser_model()