Add application file

app.py

@@ -0,0 +1,110 @@
+import math
+import numpy as np
+import pandas as pd
+import tensorflow as tf
+from tensorflow import keras
+from tensorflow.keras import layers
+import tensorflow_addons as tfa
+import matplotlib.pyplot as plt
+
+import gradio as gr
+from huggingface_hub import from_pretrained_keras
+
+model = from_pretrained_keras('keras-io/tab_transformer')
+
+CSV_HEADER = [
+    "age",
+    "workclass",
+    "fnlwgt",
+    "education",
+    "education_num",
+    "marital_status",
+    "occupation",
+    "relationship",
+    "race",
+    "gender",
+    "capital_gain",
+    "capital_loss",
+    "hours_per_week",
+    "native_country",
+    "income_bracket",
+]
+
+def get_dataset_from_pandas(data):
+    for col in data.columns:
+        if data[col].dtype == 'float64':
+            data[col] = data[col].astype('float32')
+        elif col == 'age':
+            data[col] = data[col].astype('float32')
+    ds = tf.data.Dataset.from_tensors(dict(data.drop(columns = [i for i in ['income_bracket','fnlwgt'] if i in data.columns])))
+    return ds
+
+
+def infer(age, workclass, education, education_num, marital_status, occupation, relationship, race, gender, capital_gain, capital_loss, hours_per_week, native_country):
+
+    data = pd.DataFrame({
+        'age': age,
+        'workclass': workclass,
+        'education': education,
+        'education_num': education_num,
+        'marital_status': marital_status,
+        'occupation': occupation,
+        'relationship':relationship,
+        'race': race,
+        'gender': gender,
+        'capital_gain': capital_gain,
+        'capital_loss': capital_loss,
+        'hours_per_week':hours_per_week,
+        'native_country': native_country,
+    }, index=[0])
+    validation_dataset = get_dataset_from_pandas(data)
+    # validation_dataset = get_dataset_from_csv(test_data_file, 1)
+    pred = model.predict(validation_dataset)
+
+    return f"{round(pred.flatten()[0]*100, 2)}%"
+
+# get the inputs
+inputs = [
+          gr.Slider(minimum=16, maximum=120, step=1, label='age', value=30),
+          gr.Radio(choices=[' Private', ' Local-gov', ' ?', ' Self-emp-not-inc',' Federal-gov', ' State-gov', ' Self-emp-inc', ' Without-pay', ' Never-worked'],
+                    label='workclass', type='value',value=' Private'),
+          gr.Radio(choices=[' 11th', ' HS-grad', ' Assoc-acdm', ' Some-college', ' 10th', ' Prof-school', ' 7th-8th', ' Bachelors', ' Masters', ' Doctorate',
+                            ' 5th-6th', ' Assoc-voc', ' 9th', ' 12th', ' 1st-4th', ' Preschool'],
+                   type='value', label='education', value=' Bachelors'),
+          gr.Slider(minimum=1, maximum=16, step=1, label='education_num', value=10),
+          gr.Radio(choices=[' Never-married', ' Married-civ-spouse', ' Widowed', ' Divorced', ' Separated', ' Married-spouse-absent', ' Married-AF-spouse'],
+                   type='value', label='marital_status', value=' Married-civ-spouse'),
+          gr.Radio(choices=[' Machine-op-inspct', ' Farming-fishing', ' Protective-serv', ' ?', ' Other-service', ' Prof-specialty', ' Craft-repair',
+                            ' Adm-clerical', ' Exec-managerial', ' Tech-support', ' Sales', ' Priv-house-serv', ' Transport-moving', ' Handlers-cleaners', ' Armed-Forces'],
+                   type='value', label='occupation', value=' Tech-support'),
+          gr.Radio(choices=[' Own-child', ' Husband', ' Not-in-family', ' Unmarried', ' Wife', ' Other-relative'],
+                   type='value', label='relationship', value=' Wife'),
+          gr.Radio(choices=[' Black', ' White', ' Asian-Pac-Islander', ' Other', ' Amer-Indian-Eskimo'],
+                   type='value', label='race', value=' Other'),
+          gr.Radio(choices=[' Male', ' Female'], type='value', label='gender', value=' Female'),
+          gr.Slider(minimum=0, maximum=500000, step=1, label='capital_gain', value=80000),
+          gr.Slider(minimum=0, maximum=50000, step=1, label='capital_loss', value=1000),
+          gr.Slider(minimum=1, maximum=168, step=1, label='hours_per_week', value=40),
+          gr.Radio(choices=[' United-States', ' ?', ' Peru', ' Guatemala', ' Mexico', ' Dominican-Republic', ' Ireland', ' Germany', ' Philippines', ' Thailand', ' Haiti',
+                            ' El-Salvador', ' Puerto-Rico', ' Vietnam', ' South', ' Columbia', ' Japan', ' India', ' Cambodia', ' Poland', ' Laos', ' England', ' Cuba', ' Taiwan',
+                            ' Italy', ' Canada', ' Portugal', ' China', ' Nicaragua', ' Honduras', ' Iran', ' Scotland', ' Jamaica', ' Ecuador', ' Yugoslavia', ' Hungary',
+                            ' Hong', ' Greece', ' Trinadad&Tobago', ' Outlying-US(Guam-USVI-etc)', ' France'],
+                    type='value', label='native_country', value=' Vietnam'),
+          ]
+
+# the app outputs two segmented images
+output = gr.Textbox(label='Probability of income larger than 50,000 USD per year:')
+# it's good practice to pass examples, description and a title to guide users
+title = 'Tab Transformer for Structured data'
+description = 'Using Transformer to predict whether the income will be larger than 50,000 USD given the input features.'
+
+article = "Author: <a href=\"https://huggingface.co/geninhu\">Nhu Hoang</a>. Based on this <a href=\"https://keras.io/examples/structured_data/tabtransformer/\">keras example</a> by <a href=\"https://www.linkedin.com/in/khalid-salama-24403144\">Khalid Salama.</a> HuggingFace Model <a href=\"https://huggingface.co/keras-io/tab_transformer\">here</a> "
+
+examples = [[39.0, ' Private', ' Assoc-voc', 11.0, ' Divorced', ' Tech-support', ' Not-in-family', ' White', ' Female', 0.0, 0.0, 40.0, ' United-States'],
+            [60.0, ' Private', ' 12th', 8.0, ' Married-civ-spouse', ' Handlers-cleaners', ' Husband', ' White', ' Male', 0.0, 0.0, 40.0, ' United-States'],
+            [42.0, ' Private',' Masters', 14.0, ' Married-civ-spouse', ' Prof-specialty', ' Husband', ' Asian-Pac-Islander', ' Male', 0.0, 0.0, 40.0, ' Taiwan',],
+            [31.0, ' Local-gov',' Bachelors', 13.0, ' Married-civ-spouse', ' Craft-repair', ' Husband', ' White', ' Male', 0.0, 0.0, 51.0, ' United-States'],
+            [30.0, ' Private', ' Masters', 14.0, ' Never-married', ' Prof-specialty', ' Not-in-family', ' Asian-Pac-Islander', ' Male', 0.0, 0.0, 45.0, ' Iran']]
+
+gr.Interface(infer, inputs, output, examples= examples, allow_flagging='never',
+             title=title, description=description, article=article, live=False).launch(enable_queue=True, debug=True, inbrowser=True)