Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| def make_prediction(age_first_funding_year, age_last_funding_year ,age_first_milestone_year ,age_last_milestone_year, | |
| relationships, funding_rounds, funding_total_usd, milestones, is_CA, is_NY, is_MA, is_TX, is_otherstate, | |
| is_software, is_web, is_mobile, is_enterprise, is_advertising, is_gamesvideo, is_ecommerce, is_biotech, | |
| is_consulting, is_othercategory, has_VC, has_angel, has_roundA, has_roundB, has_roundC, has_roundD, avg_participants, | |
| is_top500, has_RoundABCD, has_Investor, has_Seed, invalid_startup, tier_relationships): | |
| with open("clf.pkl", "rb") as f: | |
| clf = pickle.load(f) | |
| preds = clf.predict([[age_first_funding_year, age_last_funding_year ,age_first_milestone_year ,age_last_milestone_year, | |
| relationships, funding_rounds, funding_total_usd, milestones, is_CA, is_NY, is_MA, is_TX, is_otherstate, | |
| is_software, is_web, is_mobile, is_enterprise, is_advertising, is_gamesvideo, is_ecommerce, is_biotech, | |
| is_consulting, is_othercategory, has_VC, has_angel, has_roundA, has_roundB, has_roundC, has_roundD, avg_participants, | |
| is_top500, has_RoundABCD, has_Investor, has_Seed, invalid_startup, tier_relationships]]) | |
| if preds == 1: | |
| return "Your startup is likely to be successful..." | |
| return "Your startup is not likely to be successful" | |
| age_first_funding_year=gr.Number(label="Enter age_first_funding_year") | |
| age_last_funding_year=gr.Number(label="Enter age_last_funding_year") | |
| age_first_milestone_year=gr.Number(label="Enter age_first_milestone_year") | |
| age_last_milestone_year=gr.Number(label="Enter age_last_milestone_year") | |
| relationships=gr.Number(label="Enter relationships") | |
| funding_rounds=gr.Number(label="Enter funding_rounds") | |
| funding_total_usd=gr.Number(label="Enter funding_total_usd") | |
| milestones=gr.Number(label="Enter milestones") | |
| is_CA=gr.Number(label="is_CA") | |
| is_NY=gr.Number(label="is_CA") | |
| is_MA=gr.Number(label="is_MA") | |
| is_TX=gr.Number(label="is_MA") | |
| is_otherstate=gr.Number(label="is_otherstate") | |
| is_software=gr.Number(label="is_software") | |
| is_web=gr.Number(label="is_web") | |
| is_mobile=gr.Number(label="is_mobile") | |
| is_enterprise=gr.Number(label="is_enterprise") | |
| is_advertising=gr.Number(label="is_advertising") | |
| is_gamesvideo=gr.Number(label="is_gamesvideo") | |
| is_ecommerce=gr.Number(label="is_ecommerce") | |
| is_biotech=gr.Number(label="is_biotech=") | |
| is_consulting=gr.Number(label="is_consulting") | |
| is_othercategory=gr.Number(label="is_othercategory") | |
| has_VC=gr.Number(label="Enter has_VC") | |
| has_angel=gr.Number(label="Enter has_angel") | |
| has_roundA=gr.Number(label="Enter has_roundA") | |
| has_roundB=gr.Number(label="Enter has_roundB") | |
| has_roundC=gr.Number(label="Enter has_roundC") | |
| has_roundD=gr.Number(label="Enter has_roundD") | |
| avg_participants=gr.Number(label="Enter avg_participants") | |
| is_top500=gr.Number(label="Enter avg_participants") | |
| has_RoundABCD=gr.Number(label="Enter has_roundABCD") | |
| has_Investor=gr.Number(label="Enter has_Investor") | |
| has_Seed=gr.Number(label="Enter has_roundABCD") | |
| invalid_startup=gr.Number(label="Enter has_Seed") | |
| tier_relationships=gr.Number(label="Enter Tier_relationships") | |
| # We create the output | |
| output = gr.Textbox() | |
| app = gr.Interface(fn = make_prediction, inputs=[age_first_funding_year, age_last_funding_year ,age_first_milestone_year ,age_last_milestone_year, | |
| relationships, funding_rounds, funding_total_usd, milestones, is_CA, is_NY, is_MA, is_TX, is_otherstate, | |
| is_software, is_web, is_mobile, is_enterprise, is_advertising, is_gamesvideo, is_ecommerce, is_biotech, | |
| is_consulting, is_othercategory, has_VC, has_angel, has_roundA, has_roundB, has_roundC, has_roundD, avg_participants, | |
| is_top500, has_RoundABCD, has_Investor, has_Seed, invalid_startup, tier_relationships], outputs=output) | |
| app.launch(inline=flase) |