import gradio as gr
import pickle

# Load the pickled model
with open("../double_model.pkl", "rb") as f:
    model = pickle.load(f)

def double_number(x):
    return model.predict(x)

demo = gr.Interface(
    fn=double_number,
    inputs=gr.Number(label="Enter a number"),
    outputs=gr.Number(label="Doubled result"),
    title="🧠 Double Model",
    description="This demo takes a number and returns double using a toy model.",
)

if __name__ == "__main__":
    demo.launch()