| 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() |