from model import model
import gradio as gr
import numpy as np

def classify_image(image):
    probabilities = model.predict(image)  # Call prediction function
    return probabilities

demo = gr.Interface(
    fn=classify_image,
    inputs=gr.Image(),
    outputs=gr.Label(num_top_classes=4),
    title="Image Classifier",
    description="Upload an image and get the top 4 predicted labels with probabilities.",)
demo.launch()