| from fastai.vision.all import unet_learner, resnet34, load_learner | |
| learn = load_learner("unet_export.pkl") | |
| def segment_image(img): | |
| pred, _, _ = learn.predict(img) | |
| return pred | |
| interface = gr.Interface( | |
| fn=segment_image, | |
| inputs=gr.Image(type="pil"), | |
| outputs=gr.Image(type="numpy"), | |
| title="Segmentaci贸n Sem谩ntica con FastAI", | |
| ) | |
| interface.launch() |