import gradio as gr
import pickle

with open("model.pkl", "rb") as file:
    loaded_model = pickle.load(file)

def predict_double(number):
    # Wrap the input number into the 2D array format the model expects
    prediction = loaded_model.predict([[float(number)]])
    return prediction[0]

demo = gr.Interface(
    fn=predict_double,
    inputs=gr.Number(label="Enter a number"),
    outputs=gr.Number(label="AI Doubled Output"),
    title="Simple Linear Regression Model Test"
)

demo.launch()