import gradio as gr
import pickle

ridge_model = pickle.load(open("models/ridge.pkl","rb"))
standard_scaler = pickle.load(open("models/scaler.pkl","rb"))


def predict(temperature, rh, ws, rain, ffmc, dmc, isi, classes, region):
    new_data_scaled = standard_scaler.transform([[temperature,rh,ws,rain,ffmc,dmc,isi,classes,region]])
    result = ridge_model.predict(new_data_scaled) # result stores in list format
    return result


iface = gr.Interface(fn=predict, inputs=["text","text","text","text","text","text","text","text","text"], outputs="text")
iface.launch()