| import gradio as gr | |
| from model import model | |
| def classify_image(image): | |
| result = model.predict(image) | |
| return result | |
| demo = gr.Interface(fn=classify_image, | |
| inputs=gr.Image(), | |
| outputs=gr.Label(num_top_classes=2), | |
| title="IBM AI Capstone project: Crack detection for structural health monitoring", | |
| description="Convolutional neural network based on resnet50 model made in PyTorch to classify samples of different images of structure integrity. Demo for the final proyect of the IBM AI Engineer professional certificate. Improving AI model project THE PERFORMANCE MAY CHANGE DEPENDING OF THE USE OF THE CPU. AI model and interface built by Eng. Michael Leonardo Aguas.",) | |
| demo.launch() | |