Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| from gradio.mix import Parallel, Series | |
| import torch | |
| # Images | |
| torch.hub.download_url_to_file('http://images.cocodataset.org/val2017/000000039769.jpg', 'cat.jpg') | |
| torch.hub.download_url_to_file('https://cdn.pixabay.com/photo/2015/06/08/15/02/pug-801826_1280.jpg', 'dog.jpg') | |
| io1 = gr.Interface.load("huggingface/google/vit-base-patch16-224") | |
| io2 = gr.Interface.load("huggingface/facebook/deit-base-distilled-patch16-224") | |
| title = "VIT and Deit Parallel Demo" | |
| description = "Demo for 2 Vision Transformers: Google VIT and Facebook Deit using Gradio Parallel and Huggingface Models. Upload an image or click an example image to use. Read more at the links below" | |
| article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2010.11929'>VIT</a> | <a href='https://github.com/google-research/vision_transformer'>Vision Transformer Github</a> | <a href='https://arxiv.org/abs/2012.12877'>Deit</a>| <a href='https://github.com/facebookresearch/deit'>Deit Github</a> | <a href='https://gradio.app/blog/using-huggingface-models'>Gradio Blog</a></p>" | |
| examples = [['cat.jpg'], ['dog.jpg']] | |
| Parallel(io1, io2, title=title, description=description, article=article, examples=examples).launch() |