Update app.py

app.py

@@ -5,11 +5,11 @@ os.system("git clone https://github.com/AK391/realworld-stylegan2-encoder.git")
 os.chdir("realworld-stylegan2-encoder")
 os.system("pip install gdown")
 os.system("pip install dlib")
-os.system("gdown https://drive.google.com/uc?id=1nrz2Oc5ljvixplPra55m_NGtGAtSSEOL")
-os.system("gdown https://drive.google.com/uc?id=1WocJsMOsn6LP5NNZzhP9ErO7uz8gK8xI")
+os.system("gdown https://drive.google.com/uc?id=1i873OKcKjvpxiF0UBU4NzxlMMaD9qR5z")
 
 def inference(image):
-  os.system("python scripts/test.py --ckpt_encoder ./art_mobile_encoder_1024p.onnx --ckpt_decoder ./art_decoder_1024p.onnx --network e4e --platform onnx --size 1024 --images_path "+image.name)
+
+  os.system("python scripts/test.py --ckpt ./e4e_encode_mobile_cartoon.pt --network e4e --platform torch --size 1024 --images_path "+image.name)
   
   return "out.png"