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()