Spaces:
Runtime error
Runtime error
import gradio as gr | |
import tensorflow as tf | |
import numpy as np | |
import matplotlib.image as mpimg | |
from tensorflow.keras import models | |
model = models.load_model('alfabot.keras') | |
UPPERCASE_ALFABET = ["A", "B", "C", "D", "E", "F", "G", "H", "I", "J", "K", "L", "M", "N", "O", "P", "Q", "R", "S", "T", "U", "V", "W", "X", "Y", "Z"] | |
def predict(img): | |
image = mpimg.imread(img) | |
image = np.expand_dims(image, axis=0) | |
image_tensor = tf.constant(image, dtype=tf.float32) | |
result = model.predict(image_tensor)[0] | |
return UPPERCASE_ALFABET[np.argmax(result)] | |
gr.Interface( | |
fn=predict, | |
inputs=gr.inputs.Image(shape=(28, 28)), | |
outputs=gr.outputs.Label(num_top_classes=3), | |
title="Identificador de letras manuscritas", | |
description="Esse modelo tem a capacidade de identificar qual é a letra manuscrita.", | |
examples=["A.jpg","B.jpg","C.jpg","D.jpg","E.jpg"], | |
).launch() |