Spaces:
Runtime error
Runtime error
| import os | |
| import gradio as gr | |
| 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=1i873OKcKjvpxiF0UBU4NzxlMMaD9qR5z") | |
| os.system("wget https://github.com/kim-ninh/align_face_ffhq/raw/main/shape_predictor_68_face_landmarks.dat -P .") | |
| os.system("wget https://i.imgur.com/dJVNQSF.jpg -O ./mona.jpg") | |
| def inference(image): | |
| os.system("python scripts/test.py --align --ckpt ./e4e_encode_mobile_cartoon.pt --network e4e --platform torch --size 1024 --images_path "+image.name) | |
| return "out.png" | |
| title = "Encoder4editing" | |
| description = "Gradio demo for Encoder4editing. To use it, simply upload your image, or click one of the examples to load them. Read more at the links below." | |
| article = "<p style='text-align: center'><a href='https://github.com/TalkUHulk/realworld-stylegan2-encoder' target='_blank'>Github Repo</a></p>" | |
| gr.Interface( | |
| inference, | |
| gr.inputs.Image(type="file", label="Input"), | |
| gr.outputs.Image(type="file", label="Output"), | |
| title=title, | |
| description=description, | |
| article=article, | |
| enable_queue=True, | |
| examples=[['mona.jpg']] | |
| ).launch(debug=True) |