ashishabraham22's picture
Upload app.py
94d7100
raw history blame
No virus
373 Bytes
import gradio as gr
import numpy as np
from ml_predict import preprocess_img,predict_defect
image = gr.inputs.Image(shape=(256, 256))
label = gr.outputs.Label(num_top_classes=3)
gr.Interface(fn=predict_defect, inputs=image, outputs=label, theme='peach',examples=[["images\Cr2.bmp"], ["images\Pa2.bmp"], ["images\scratch1.bmp"], ["images\Pit2.bmp"]]).launch()