Update app.py

app.py

@@ -9,7 +9,6 @@ import gradio as gr
 import matplotlib.pyplot as plt
 import seaborn as sns
 import io
-import base64
 from PIL import Image
 import warnings
 warnings.filterwarnings('ignore')
@@ -210,8 +209,7 @@ with gr.Blocks() as demo:
     train_button.click(
         fn=train_model,
         inputs=[file_input, n_estimators, learning_rate, max_depth, subsample],
-        outputs=[output_text, accuracy_plot, confusion_plots],
-        _js="() => {return {progress: true}}"
+        outputs=[output_text, accuracy_plot, confusion_plots]
     )
 
 demo.launch()