import gradio as gr
import pickle
import double_model  # 👈 this must be imported first!

# Now we can safely unpickle
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="Enter a number and get double. Just a toy demo model.",
)

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