Spaces:
Runtime error
Runtime error
| import pickle | |
| import gradio as gr | |
| def make_prediction(age1,gender1,occupation1,line_of_work1,time_bp1,time_dp1,travel_time1,easeof_online1,home_env1,prod_inc1,sleep_bal1,new_skill1,fam_connect1,relaxed1,self_time1,like_hw1,dislike_hw1,certaindays_hw1): | |
| #print(age1+gender1+occupation1+line_of_work1+time_bp1+time_dp1+travel_time1+easeof_online1+home_env1+prod_inc1+sleep_bal1+new_skill1+fam_connect1+relaxed1+self_time1+like_hw1+dislike_hw1+certaindays_hw1) | |
| print(age1+gender1+occupation1+line_of_work1+certaindays_hw1) | |
| if age1=="18": | |
| age1=int(1) | |
| elif age1=="19-25": | |
| age1=int(0) | |
| elif age1=="33-40": | |
| age1=int(2) | |
| elif age1=="60+": | |
| age1=int(3) | |
| elif age1=="26-32": | |
| age1=int(4) | |
| elif age1=="40-50": | |
| age1=int(5) | |
| else: age1=int(6) | |
| print(str(age1)) | |
| if gender1=="Male": | |
| gender1=int(0) | |
| elif gender1=="Female": | |
| gender1=int(1) | |
| else: gender1=int(2) | |
| print(str(gender1)) | |
| if occupation1=="Student in College": | |
| occupation1=int(0) | |
| elif occupation1=="Student in School": | |
| occupation1=int(1) | |
| elif occupation1=="Working Professional": | |
| occupation1=int(2) | |
| elif occupation1=="Entrepreneur": | |
| occupation1=int(3) | |
| elif occupation1=="Retired/Senior Citizen": | |
| occupation1=int(4) | |
| elif occupation1=="Homemaker": | |
| occupation1=int(5) | |
| else: occupation1=int(6) | |
| print(str(occupation1)) | |
| if line_of_work1=="Teaching": | |
| line_of_work1=int(0) | |
| elif line_of_work1=="Engineering": | |
| line_of_work1=int(1) | |
| elif line_of_work1=="Management": | |
| line_of_work1=int(2) | |
| elif line_of_work1=="APSPDCL": | |
| line_of_work1=int(3) | |
| elif line_of_work1=="Architecture": | |
| line_of_work1=int(4) | |
| elif line_of_work1=="Other": | |
| line_of_work1=int(5) | |
| else: line_of_work1=int(6) | |
| print(str(line_of_work1)) | |
| if certaindays_hw1=="Yes": | |
| certaindays_hw1=int(0) | |
| elif certaindays_hw1=="No": | |
| certaindays_hw1=int(1) | |
| else: certaindays_hw1=int(2) | |
| print(str(certaindays_hw1)) | |
| with open("covid_psyc_model.pkcls", "rb") as f: | |
| #Then feeds our data into the model, then sets the "preds" variable to the prediction output for our class variable, which is price. | |
| clf = pickle.load(f) | |
| preds=clf.predict([[age1,gender1,occupation1,line_of_work1,time_bp1,time_dp1,travel_time1,easeof_online1,home_env1,prod_inc1,sleep_bal1,new_skill1,fam_connect1,relaxed1,self_time1,like_hw1,dislike_hw1,certaindays_hw1]]) | |
| if preds == 0: | |
| return "Complete Physical Attendance" | |
| elif preds== 1: | |
| return "Work/study from home" | |
| else: | |
| return "Please check and re-enter your inputs" | |
| #Finally, we send the prediction to the website. | |
| return preds | |
| HasAge=gr.Dropdown(["18","19-25","26-32","33-40","40-50","50-60","60+"],label="Please select your age range") | |
| HasGender=gr.Dropdown(["Male","Female","Prefer not to say"],label="Please select your gender") | |
| HasOccupation=gr.Dropdown(["Student in College", "Student in School", "Working Professional", | |
| "Entrepreneur", "Retired/Senior Citizen", "Homemaker", | |
| "Currently Out of Work", | |
| "Medical Professional aiding efforts against COVID-19"],label="Please select your occupation") | |
| HasLineOfWork=gr.Dropdown(["Teaching","Engineering","Management","APSPDCL","Architecture","Other","Government Employee"],label="Please select your line of work") | |
| HasTimeBP=gr.Slider(minimum=4,maximum=12,step=1,label="How many hours before pandemic did the employee work?") | |
| HasTimeDP=gr.Slider(minimum=4,maximum=12,step=1,label="How many hours after pandemic did the employee work?") | |
| HasTravelTime=gr.Slider(minimum=0.5,maximum=3,step=0.5,label="How many hours does the employee travel for work?") | |
| HasEaseOfOnline=gr.Slider(minimum=1,maximum=5,step=1,label="On a scale of 1-5, how easy does the employee find to work online?") | |
| HasHomeEnv=gr.Slider(minimum=1,maximum=5,step=1,label="On a scale of 1-5, how comfortable is the employee in home environment?") | |
| HasProdInc=gr.Slider(minimum=-1,maximum=1,step=0.5,label="On a scale of -1 to 1, how productive is the employee?") | |
| HasSleepBal=gr.Slider(minimum=-1,maximum=1,step=0.5,label="On a scale of -1 to 1, how is the employee's sleep cycle?") | |
| HasNewSkill=gr.Slider(minimum=-1,maximum=1,step=0.5,label="On a scale of -1 to 1, how likely is it that the employee learnt a new skill?") | |
| HasFamConnect=gr.Slider(minimum=-1,maximum=1,step=0.5,label="On a scale of -1 to 1, how is the employee's family connection impacted?") | |
| HasRelaxed=gr.Slider(minimum=-1,maximum=1,step=0.5,label="On a scale of -1 to 1, how relaxed does the employee feel?") | |
| HasSelfTime=gr.Slider(minimum=-1,maximum=1,step=0.5,label="On a scale of -1 to 1, how likely does the employee feel the presence of self-time?") | |
| HasLikeHW=gr.Slider(minimum=0,maximum=1,step=0.1,label="On a scale of 0 to 1, how much does the employee like working from home?") | |
| HasDislikeHW=gr.Slider(minimum=0,maximum=1,step=0.1,label="On a scale of 0 to 1, how much does the employee dislike working from home?") | |
| HasCertainDaysHW=gr.Dropdown(["Yes","No","Maybe"],label="Is the employee okay to work in a hybrid setting?") | |
| output = gr.Textbox(label="Employee Preference Prediction:") | |
| app = gr.Interface(fn = make_prediction, inputs=[HasAge,HasGender,HasOccupation,HasLineOfWork,HasTimeBP,HasTimeDP,HasTravelTime,HasEaseOfOnline,HasHomeEnv,HasProdInc,HasSleepBal,HasNewSkill,HasFamConnect,HasRelaxed,HasSelfTime,HasLikeHW,HasDislikeHW,HasCertainDaysHW], outputs=output) | |
| app.launch() |