File size: 790 Bytes
2f89dbe |
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 |
import gradio as gr
import pathlib
current_dir = pathlib.Path(__file__)
images = [current_dir / "cheetah1.jpeg", current_dir / "cheetah1.jpg", current_dir / "lion.jpg"]
img_classifier = gr.Interface.load(
"models/google/vit-base-patch16-224", examples=images, cache_examples=True
)
def func(img, text):
return img_classifier(img), text
using_img_classifier_as_function = gr.Interface(
func,
[gr.Image(type="filepath"), "text"],
["label", "text"],
examples=[
[current_dir / "cheetah1.jpeg", None],
[current_dir / "cheetah1.jpg", "cheetah"],
[current_dir / "lion.jpg", "lion"],
],
cache_examples=True,
)
demo = gr.TabbedInterface([using_img_classifier_as_function, img_classifier])
if __name__ == "__main__":
demo.launch()
|