Gabriel commited on
Commit
c48b9e5
1 Parent(s): 130eebb

feat: model, app, req

Browse files
Files changed (4) hide show
  1. .gitattributes +1 -0
  2. app.py +39 -0
  3. model.pkl +3 -0
  4. requirements.txt +1 -0
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ model.pkl filter=lfs diff=lfs merge=lfs -text
app.py ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import torch
3
+ from torch import tensor
4
+ from torch.nn import functional as F
5
+ from sklearn.preprocessing import LabelEncoder
6
+ import pandas as pd
7
+
8
+ label_encoder = LabelEncoder()
9
+
10
+ model = torch.load('model.pkl')
11
+
12
+ def calc(X_test):
13
+ # return model.predict(X_test)
14
+ return X_test
15
+
16
+ def main(inputs):
17
+ df = pd.DataFrame(columns=inputs[1:])
18
+ df.loc[0] = inputs[1:]
19
+
20
+ for column in df.columns:
21
+ df[column] = label_encoder.fit_transform(df[column])
22
+
23
+ t_indep = tensor(df[indep_cols].values, dtype=torch.float)
24
+ vals,indices = t_indep.max(dim=0)
25
+ t_indep = t_indep / vals
26
+ return calc_preds(coeffs, t_indep)
27
+
28
+ iface = gr.Interface(
29
+ fn=calc,
30
+ inputs=[gr.DropDown('Feature 1', choices=['Option 1', 'Option 2', 'Option 3']),
31
+ gr.DropDown('Feature 2', choices=['Option A', 'Option B', 'Option C']),
32
+ # Add more DropDown or other input components as needed
33
+ ],
34
+ outputs="text",
35
+ title="Heart Disease identifier",
36
+ description="Identifies if a person has/will have a heart disease"
37
+ )
38
+
39
+ iface.launch()
model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a72df222ca0d5752241edf45b1670fde6d6391e047d0095ad24d5485bde9f4eb
3
+ size 193440658
requirements.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ fastai