Spaces:
Runtime error
Runtime error
import numpy as np | |
import gradio as gr | |
import tensorflow as tf | |
model = tf.keras.models.load_model('ev2m') | |
classes = np.loadtxt('classes.txt', dtype=str) | |
classes.sort() | |
classes = list(classes[:20]) | |
classes.append('other') | |
def pred(image): | |
image = np.array(image) / 255.0 | |
image = tf.image.resize(image, (480, 480)) | |
image = tf.keras.preprocessing.image.img_to_array(image) | |
pred = model.predict(np.expand_dims(image, axis=0)) | |
return {classes[i]: float(pred[0][i]) for i in range(len(classes))} | |
demo = gr.Interface(fn=pred, inputs="image", outputs="label") | |
demo.launch() |