ArchiMathur commited on
Commit
be07aa4
·
verified ·
1 Parent(s): 588c0b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -7,6 +7,8 @@ import numpy as np
7
  import pickle
8
  import sklearn
9
  from datasets import load_dataset
 
 
10
 
11
  # Read the data
12
  data = pd.read_csv("mldata.csv")
@@ -167,7 +169,7 @@ worker_list = ["hard worker", "smart worker"]
167
  demo = gr.Interface(
168
  fn=rfprediction,
169
  inputs=[
170
- gr.Dropdown(["Random Forest", "Decision Tree"], label="Select Machine Learning Model"),
171
  gr.Textbox(placeholder="What is your name?", label="Name"),
172
  gr.Slider(minimum=1, maximum=9, value=3, step=1, label="Are you a logical thinking person?", info="Scale: 1 - 9"),
173
  gr.Slider(minimum=0, maximum=6, value=0, step=1, label="Do you attend any Hackathons?", info="Scale: 0 - 6 | 0 - if not attended any"),
 
7
  import pickle
8
  import sklearn
9
  from datasets import load_dataset
10
+ Copyimport joblib
11
+ from tensorflow.keras.models import load_model
12
 
13
  # Read the data
14
  data = pd.read_csv("mldata.csv")
 
169
  demo = gr.Interface(
170
  fn=rfprediction,
171
  inputs=[
172
+ gr.Dropdown(["Random Forest", "Decision Tree","Sequential Model"], label="Select Machine Learning Model"),
173
  gr.Textbox(placeholder="What is your name?", label="Name"),
174
  gr.Slider(minimum=1, maximum=9, value=3, step=1, label="Are you a logical thinking person?", info="Scale: 1 - 9"),
175
  gr.Slider(minimum=0, maximum=6, value=0, step=1, label="Do you attend any Hackathons?", info="Scale: 0 - 6 | 0 - if not attended any"),