Spaces:
Runtime error
Runtime error
| import tensorflow as tf | |
| import gradio as gr | |
| import numpy as np | |
| modelo =tf.keras.models.load_model('model_0.h5') | |
| #modelo.load_weights("model_0_weights.h5") | |
| classes=['daisy','dandelion','roses','sunflowers','tulips'] | |
| def classifier(image): | |
| pred_img = modelo.predict(tf.expand_dims(image,axis=0)) | |
| pred_img = tf.squeeze(tf.round(pred_img)) | |
| texto = str(f'Predicted label: {classes[(np.argmax(pred_img))]}!!!') | |
| return texto | |
| interface = gr.Interface(classifier,gr.inputs.Image(shape=(180,180)),outputs = "text", | |
| description="Classifier of images of daisy plants, dandelion, roses, sunflowers, and tulips", | |
| title="Flower Image Classifier", | |
| examples=[['1022552036_67d33d5bd8_n.jpg'],['142218310_d06005030a_n.jpg'], | |
| ['2077865117_9ed85191ae_n.jpg'],['3500121696_5b6a69effb_n.jpg'],['4675287055_5938ed62c4.jpg']]) | |
| interface.launch() | |