File size: 1,213 Bytes
a1dc0fd 45990d1 a1dc0fd 45990d1 a1dc0fd 45990d1 a1dc0fd 45990d1 a1dc0fd 45990d1 a1dc0fd 45990d1 a1dc0fd 45990d1 a1dc0fd 45990d1 |
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 34 35 |
import gradio as gr
import tensorflow as tf
from tensorflow.keras.preprocessing import image
import numpy as np
# Load the Teachable Machine Image Model
model_url = "https://teachablemachine.withgoogle.com/models/ZPfAhDYCh/"
model = tf.keras.models.load_model(model_url + "model.json")
# Function to preprocess the image before making predictions
def preprocess_image(img):
img = image.img_to_array(img)
img = np.expand_dims(img, axis=0)
img /= 255.0 # Normalize pixel values to be between 0 and 1
return img
# Function to make predictions using the loaded model
def predict_image(img):
img = preprocess_image(img)
prediction = model.predict(img)
return {class_name: float(prediction[0][i]) for i, class_name in enumerate(classes)}
# Fetch class labels from metadata.json
metadata_url = model_url + "metadata.json"
metadata = tf.keras.utils.get_file("metadata.json", metadata_url)
classes = tf.keras.utils.get_file("classes.txt", metadata_url.replace("metadata.json", "classes.txt")).read().splitlines()
# Create Gradio interface
iface = gr.Interface(fn=predict_image, inputs="image", outputs="label", live=True, capture_session=True)
# Launch Gradio interface
iface.launch()
|