Viet
add application file and sample images
391a220
raw
history blame
334 Bytes
# an image classification demo!
import torch
import gradio as gr
model1 = gr.Interface.load("huggingface/microsoft/beit-base-patch16-224")
model2 = gr.Interface.load("huggingface/google/vit-base-patch16-224")
gr.Parallel(model1, model2,
examples=['cat.jpg', 'dog.jpg', 'alligator knife head.jpg', 'moskva.jpg']).launch()