3v324v23 commited on
Commit
6470c0e
·
1 Parent(s): 2faf709

spinner transformer

Browse files
Files changed (1) hide show
  1. streamlit_simulation/app.py +2 -1
streamlit_simulation/app.py CHANGED
@@ -486,7 +486,8 @@ if model_choice == "Transformer Model (moments)":
486
  plot_title, plot_container, x_axis_label, info_container = init_simulation_layout()
487
 
488
  # Zugriff auf Modell, Dataset, Device
489
- model, test_dataset, device = load_transformer_model_and_dataset()
 
490
  data = test_dataset.data # bereits skaliert
491
  scaler = test_dataset.scaler
492
  n_channels = test_dataset.n_channels
 
486
  plot_title, plot_container, x_axis_label, info_container = init_simulation_layout()
487
 
488
  # Zugriff auf Modell, Dataset, Device
489
+ with st.spinner("Initializing Transformer model please wait..."):
490
+ model, test_dataset, device = load_transformer_model_and_dataset()
491
  data = test_dataset.data # bereits skaliert
492
  scaler = test_dataset.scaler
493
  n_channels = test_dataset.n_channels