File size: 1,632 Bytes
2543307 869d204 d6be5e1 2543307 03b9368 6402598 2543307 03b9368 2543307 03b9368 2543307 41b8a6e 2543307 1906885 2543307 ffda472 2543307 1643b12 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
import os
import gradio as gr
import sys
os.system("pip install gradio==2.4.6")
os.system("git clone https://github.com/AK391/PAMA.git")
os.chdir('PAMA')
os.system("wget https://upload.wikimedia.org/wikipedia/commons/thumb/1/1e/Hoover_Tower_Stanford_January_2013.jpg/800px-Hoover_Tower_Stanford_January_2013.jpg")
os.system("wget https://upload.wikimedia.org/wikipedia/commons/thumb/e/ea/Van_Gogh_-_Starry_Night_-_Google_Art_Project.jpg/1920px-Van_Gogh_-_Starry_Night_-_Google_Art_Project.jpg")
os.system("gdown https://drive.google.com/uc?id=1rPB_qnelVVSad6CtadmhRFi0PMI_RKdy")
os.mkdir("checkpoints/")
os.system("unzip -j original_PAMA.zip -d checkpoints/")
def inference(content,style):
os.system("python main.py eval --content "+content+" --style "+style)
return "ics.jpg"
title = "PAMA"
description = "Gradio demo for PAMA: Consistent Style Transfer. To use it, simply upload your images, or click one of the examples to load them. Read more at the links below."
article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2201.02233' target='_blank'>Consistent Style Transfer</a> | <a href='https://github.com/computer-vision2022/PAMA' target='_blank'>Github Repo</a></p>"
examples=[['800px-Hoover_Tower_Stanford_January_2013.jpg','1920px-Van_Gogh_-_Starry_Night_-_Google_Art_Project.jpg']]
gr.Interface(
inference,
[gr.inputs.Image(type="filepath", label="Content"),gr.inputs.Image(type="filepath", label="Style")],
gr.outputs.Image(type="file", label="Output"),
title=title,
description=description,
article=article,
examples=examples,
enable_queue=True
).launch() |