Update pages/gpt.py

pages/gpt.py

@@ -4,46 +4,6 @@ import torch
 import textwrap
 import plotly.express as px
 
-# df = px.data.iris()
-
-# @st.cache_data
-# def get_img_as_base64(file):
-#     with open(file, "rb") as f:
-#         data = f.read()
-#     return base64.b64encode(data).decode()
-
-# page_bg_img = f"""
-# <style>
-# [data-testid="stAppViewContainer"] > .main {{
-# background-image: url("https://i.pinimg.com/originals/9f/57/bd/9f57bd45d33eb906fdb3d7ffe22e2058.png");
-# background-size: 100%;
-# background-position: top left;
-# background-repeat: no-repeat;
-# background-attachment: local;
-# }}
-# # [data-testid="stSidebar"] > div:first-child {{
-# # background-image: url("https://catherineasquithgallery.com/uploads/posts/2021-02/1614542041_37-p-fon-belii-tekstura-43.jpg");
-# # background-size: 100%;
-# # background-position: center; 
-# # background-repeat: no-repeat;
-# # background-attachment: fixed;
-# # }}
-# [data-testid="stHeader"] {{
-# background: rgba(0,0,0,0);
-# }}
-# [data-testid="stToolbar"] {{
-# right: 2rem;
-# }}
-# div.css-1n76uvr.esravye0 {{
-# background-color: rgba(238, 238, 238, 0.5);
-# border: 10px solid #EEEEEE;
-# padding: 5% 5% 5% 10%;
-# border-radius: 5px;
-# }}
-# </style>
-# """
-
-# st.markdown(page_bg_img, unsafe_allow_html=True)
 
 st.markdown('## Генерация текста GPT-моделью')
 
@@ -82,6 +42,9 @@ if st.button('Сгенерировать текст'):
             no_repeat_ngram_size=3,
             num_return_sequences=num_samples,
             ).cpu().numpy()
+        st.write('**_Результат_** 👇')
         for i, out_ in enumerate(out):
-            st.write(f'Текст {i+1}:')
-            st.write(textwrap.fill(tokenizer.decode(out_), 100))
+            
+            with st.expander(f'Текст {i+1}:'):
+                st.write(textwrap.fill(tokenizer.decode(out_), 100))
+                st.image("pict/wow.png")