crop_yield / app.py
abdulsamod's picture
Update app.py
79cdaa6 verified
import pickle
import numpy as np
import gradio as gr
import sklearn
filename= "model.pkl"
model_ = pickle.load(open(filename, 'rb'))
def crop_yield(Planted,Size_of_land, Bags, Soil_Type, Type_of_Seed, Color_of_Seeds):
prediction = model_.predict([[Planted,Size_of_land, Bags, Soil_Type, Type_of_Seed,Color_of_Seeds]])
return prediction
#create input and output objects
#input object1
input1 = gr.inputs.Number(label="Planted")
#input object 2
input2 = gr.inputs.Number(label="Size of land (Hectare)")
#input object3
input3 = gr.inputs.Number(label="Number of Bags")
#input object 3
input4 = gr.inputs.Number(label="Soil Type Input: 0 for Loamy Soil and 1 for others")
#input object 3
input5 = gr.inputs.Number(label="Type of Seed Input: 0 for Super Seed and 1 for others")
#input object 3
input6 = gr.inputs.Number(label="Color of Seeds Input: 0 for White Seed and 1 for others")
#output object
output = gr.outputs.Textbox(label= "Number of Yields")
#create interface
gui = gr.Interface(fn=crop_yield,
inputs=[input1, input2, input3, input4, input5, input6],
outputs=output,
description="Here's a simple webapp to calculate the number of possible yield.").launch()