lab1 / main.py
vrentati's picture
Upload 2 files
bcc0ae1 verified
raw
history blame contribute delete
637 Bytes
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()