Update pages/gpt.py

pages/gpt.py

@@ -32,7 +32,8 @@ top_p = st.sidebar.slider('**Minimum total probability of top words:**', 0.4, 1.
 
 prompt = st.text_input('**Enter text 👇:**')
 if st.button('**Generate text**'):
- 
+    image_container = st.empty()
+    image_container.image("pict/wait.png", caption="that's so long!!!", use_column_width=True)
     with torch.inference_mode():
         prompt = tokenizer.encode(prompt, return_tensors='pt')
         out = model.generate(
@@ -46,6 +47,7 @@ if st.button('**Generate text**'):
             no_repeat_ngram_size=3,
             num_return_sequences=num_samples,
             ).cpu().numpy()
+        image_container.empty()
         st.write('**_Результат_** 👇')
         for i, out_ in enumerate(out):
             # audio_file = open('pict/pole-chudes-priz.mp3', 'rb')