adf0d17497
publish / version_or_publish (push) Waiting to run
storybook-build / changes (push) Waiting to run
storybook-build / :storybook-build (push) Blocked by required conditions
Sync Gradio Skills to Hugging Face / sync-skills (push) Waiting to run
functional / changes (push) Waiting to run
functional / build-frontend (push) Blocked by required conditions
functional / functional-test-SSR=false (push) Blocked by required conditions
functional / functional-reload (push) Blocked by required conditions
functional / functional-test-SSR=true (push) Blocked by required conditions
hygiene / hygiene-test (push) Waiting to run
python / changes (push) Waiting to run
python / build (push) Blocked by required conditions
python / test-ubuntu-latest-flaky (push) Blocked by required conditions
python / test-ubuntu-latest-not-flaky (push) Blocked by required conditions
python / test-windows-latest-flaky (push) Blocked by required conditions
python / test-windows-latest-not-flaky (push) Blocked by required conditions
js / changes (push) Waiting to run
js / js-test (push) Blocked by required conditions
docs-build / changes (push) Waiting to run
docs-build / docs-build (push) Blocked by required conditions
docs-build / website-build (push) Blocked by required conditions
31 lines
839 B
Python
31 lines
839 B
Python
import gradio as gr
|
|
import pathlib
|
|
|
|
current_dir = pathlib.Path(__file__).parent
|
|
|
|
images = [str(current_dir / "cheetah1.jpeg"), str(current_dir / "cheetah1.jpg"), str(current_dir / "lion.jpg")]
|
|
|
|
img_classifier = gr.load(
|
|
"models/google/vit-base-patch16-224", examples=images, cache_examples=False
|
|
)
|
|
|
|
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=[
|
|
[str(current_dir / "cheetah1.jpeg"), None],
|
|
[str(current_dir / "cheetah1.jpg"), "cheetah"],
|
|
[str(current_dir / "lion.jpg"), "lion"],
|
|
],
|
|
cache_examples=False,
|
|
api_name="predict"
|
|
)
|
|
demo = gr.TabbedInterface([using_img_classifier_as_function, img_classifier])
|
|
|
|
if __name__ == "__main__":
|
|
demo.launch()
|