| import pathlib | |
| from fastai.vision.all import load_learner | |
| import gradio as gr | |
| from pathlib import Path | |
| import pathlib | |
| temp = pathlib.PosixPath | |
| pathlib.PosixPath = pathlib.WindowsPath | |
| # Завантажуємо модель (виправляємо шлях) | |
| learn = load_learner(Path("model.pkl")) | |
| # Функція для класифікації | |
| def classify_image(img): | |
| pred, _, probs = learn.predict(img) | |
| return f"Prediction: {pred}; Probability: {probs.max():.04f}" | |
| # Інтерфейс Gradio | |
| gr.Interface( | |
| fn=classify_image, | |
| inputs=gr.Image(type="pil"), | |
| outputs=gr.Text() | |
| ).launch() | |