Update app.py

app.py

@@ -7,8 +7,6 @@ import pandas as pd
 import gradio as gr
 import urllib
 
-
-
 input_path = "https://archive.ics.uci.edu/ml/machine-learning-databases/census-income-mld/census-income"
 input_column_header = "income_level"
 
@@ -56,7 +54,9 @@ feature_d = tfdf.keras.FeatureUsage(name="capital_losses", semantic=tfdf.keras.F
 train_ds = tfdf.keras.pd_dataframe_to_tf_dataset(train_data, label="income_level")
 test_ds = tfdf.keras.pd_dataframe_to_tf_dataset(test_data, label="income_level")
 
-tf.autograph.set_verbosity(0)
+import logging
+logging.getLogger("tensorflow").setLevel(logging.ERROR)
+
 # Train a GB Trees model
 model = tfdf.keras.GradientBoostedTreesModel(
       features = [feature_a, feature_b, feature_c, feature_d],