Ignahugging's picture
Update app.py
5988d6c
raw
history blame contribute delete
No virus
939 Bytes
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()