geninhu commited on
Commit
73ffb7a
·
1 Parent(s): 2c710ef

Update application file

Browse files
Files changed (1) hide show
  1. app.py +0 -20
app.py CHANGED
@@ -4,32 +4,12 @@ import pandas as pd
4
  import tensorflow as tf
5
  from tensorflow import keras
6
  from tensorflow.keras import layers
7
- import tensorflow_addons as tfa
8
- import matplotlib.pyplot as plt
9
 
10
  import gradio as gr
11
  from huggingface_hub import from_pretrained_keras
12
 
13
  model = from_pretrained_keras('keras-io/tab_transformer')
14
 
15
- CSV_HEADER = [
16
- "age",
17
- "workclass",
18
- "fnlwgt",
19
- "education",
20
- "education_num",
21
- "marital_status",
22
- "occupation",
23
- "relationship",
24
- "race",
25
- "gender",
26
- "capital_gain",
27
- "capital_loss",
28
- "hours_per_week",
29
- "native_country",
30
- "income_bracket",
31
- ]
32
-
33
  def get_dataset_from_pandas(data):
34
  for col in data.columns:
35
  if data[col].dtype == 'float64':
 
4
  import tensorflow as tf
5
  from tensorflow import keras
6
  from tensorflow.keras import layers
 
 
7
 
8
  import gradio as gr
9
  from huggingface_hub import from_pretrained_keras
10
 
11
  model = from_pretrained_keras('keras-io/tab_transformer')
12
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13
  def get_dataset_from_pandas(data):
14
  for col in data.columns:
15
  if data[col].dtype == 'float64':