| from keras.models import load_model | |
| import numpy as np | |
| from keras.preprocessing import image | |
| import gradio as gr | |
| from PIL import Image | |
| def show(img): | |
| img = img.reshape( 64,64,3) | |
| model=load_model('./demo.h5') | |
| test_image=np.expand_dims(img, axis=0) | |
| result=model.predict(test_image) | |
| if result[0][0]==1: | |
| prediction='Dog' | |
| print(prediction) | |
| return prediction | |
| else: | |
| prediction='Cat' | |
| print(prediction) | |
| return prediction | |
| input = gr.inputs.Image(type='pil', label="Original Image", source="upload", optional=True) | |
| inputs = [input] | |
| outputs = gr.outputs.Image(type="pil", label="Output Image") | |
| title = "Dog and Cat Image Classification" | |
| image = gr.inputs.Image(shape=(64,64)) | |
| demo=gr.Interface(fn=show, inputs=image,examples=["photo/a01.jpg", "photo/a02.jpg","photo/a03.jpg","photo/a04.jpg"],title=title,outputs="text").launch(debug='True') | |
| if __name__ == "__main__": | |
| demo.launch() |