waiting msg while generating

app.py

@@ -9,7 +9,6 @@ seq_num = st.number_input('Number of sentences to generate ', 1, 20, 5)
 max_len = st.number_input('Length of the sentence ', 5, 300, 100)
 
 go = st.button('Generate')
-
 with st.spinner('Waiting for the model to load.....'):
     model = AutoModelForCausalLM.from_pretrained('flax-community/Sinhala-gpt2')
     tokenizer = AutoTokenizer.from_pretrained('flax-community/Sinhala-gpt2')
@@ -18,8 +17,9 @@ st.success('Model loaded!!')
 
 if go:
     try:
-        generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
-        seqs = generator(seed, max_length=max_len, num_return_sequences=seq_num)
+        with st.spinner('Generating...'):
+            generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
+            seqs = generator(seed, max_length=max_len, num_return_sequences=seq_num)
         st.write(seqs)
     except Exception as e:
         st.exception(f'Exception: {e}')