import gradio as gr
from predict import predict

demo = gr.Interface(
    fn=predict,
    inputs=gr.Image(
        type="numpy",
        label="Upload Cat/Dog Image"
    ),
    outputs=gr.Label(num_top_classes=2, label="Prediction"),
    title="🐱 Cat vs 🐶 Dog Classifier",
    description="Upload an image to classify it as a cat or dog."
)

demo.launch()