Upload app.py

app.py

@@ -2,13 +2,14 @@ import math
 import numpy as np
 import pandas as pd
 import tensorflow as tf
+import tensorflow_addons as tfa
 from tensorflow import keras
 from tensorflow.keras import layers
 
 import gradio as gr
 from huggingface_hub import from_pretrained_keras
 
-model = from_pretrained_keras('keras-io/tab_transformer')
+model = from_pretrained_keras('keras-io/tab_transformer', custom_objects={'optimizer': tfa.optimizers.AdamW})
 
 def get_dataset_from_pandas(data):
     for col in data.columns:
@@ -87,4 +88,4 @@ examples = [[39.0, ' Private', ' Assoc-voc', 11.0, ' Divorced', ' Tech-support',
             [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=False)
+             title=title, description=description, article=article, live=False).launch(enable_queue=True, debug=True, inbrowser=True)