ForestFire / app.py
totoshi's picture
Upload 3 files
970e332 verified
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()