mirror/gradio
2
0
Fork 0
mirror of https://github.com/gradio-app/gradio.git synced 2024-11-27 01:40:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

add imagenet labels for demo

Browse Source
This commit is contained in:
Ali Abid 2020-11-02 18:21:31 -08:00
parent 37a552c9c1
commit 9f9991d058
2 changed files with 1004 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1000
demo/files/imagenet_labels.json Normal file
View File

File diff suppressed because it is too large Load Diff

7
demo/image_classifier.py
View File

@ -7,10 +7,11 @@ import numpy as np
from PIL import Image
import requests
from urllib.request import urlretrieve
import json
# # Download human-readable labels for ImageNet.
response = requests.get("https://git.io/JJkYN")
labels = response.text.split("\n")
# Load human-readable labels for ImageNet.
with open("files/imagenet_labels.json") as labels_file:
labels = json.load(labels_file)
mobile_net = tf.keras.applications.MobileNetV2()