|
import gradio as gr |
|
import tensorflow as tf |
|
import numpy as np |
|
from PIL import Image |
|
model_path = "Xception.keras" |
|
model = tf.keras.models.load_model(model_path) |
|
|
|
def predict_pokemon(image): |
|
|
|
print(type(image)) |
|
image = Image.fromarray(image.astype('uint8')) |
|
image = image.resize((150, 150)) |
|
image = np.array(image) |
|
image = np.expand_dims(image, axis=0) |
|
|
|
|
|
prediction = model.predict(image) |
|
|
|
|
|
|
|
prediction = np.round(prediction, 2) |
|
|
|
P_aloevera = prediction[0][0] |
|
P_curcuma = prediction[0][1] |
|
p_guava = prediction[0][2] |
|
return {'aloevera': P_aloevera, 'curcuma': P_curcuma, 'guava': p_guava} |
|
|
|
input_image = gr.Image() |
|
iface = gr.Interface( |
|
fn=predict_pokemon, |
|
inputs=input_image, |
|
outputs=gr.Label(), |
|
examples=["images/aloevera0.jpg", "images/curcuma51.jpg", "images/guava10.jpg"], |
|
description="A simple mlp classification model for image classification using the mnist dataset.") |
|
iface.launch(share=True) |