app.py

import gradio as gr
from fastai.vision.all import *
from huggingface_hub import from_pretrained_fastai
import platform
import pathlib
import sys

# --- PARCHE 1: Rutas de Windows ---
if platform.system() != 'Windows':
    pathlib.WindowsPath = pathlib.PosixPath

# --- PARCHE 2: El fantasma de Plum ---
# Como el modelo está mezclado, cuando pida el archivo viejo ('plum.function'),
# le damos la librería nueva ('plum') para que no explote.
try:
    import plum
    sys.modules['plum.function'] = plum
except ImportError:
    pass

# --- Código Normal ---
repo_id = "alperema/practica1-neumonia"
learner = from_pretrained_fastai(repo_id)
labels = learner.dls.vocab

def predict(img):
    img = PILImage.create(img)
    pred, pred_idx, probs = learner.predict(img)
    return {labels[i]: float(probs[i]) for i in range(len(labels))}

gr.Interface(
    fn=predict, 
    inputs=gr.Image(), 
    outputs=gr.Label(num_top_classes=2),
    examples=['normal.jpeg', 'neumonia.jpeg'] 
).launch()