Update app.py

app.py

@@ -107,7 +107,9 @@ with st.sidebar:
 st.title("text")
 
 if submitted:
-    trainer = Trainer(model_configs[model_name])
-    trainer.inference()
+    with st.spinner('Creating the trainer class instance...'):
+        trainer = Trainer(model_configs[model_name])
+    with st.spinner('Running inference function...')
+        trainer.inference()
     st.success(f"Success with the inference of {model_name}")