Update pages/gpt.py

pages/gpt.py

@@ -4,6 +4,14 @@ import torch
 import textwrap
 import plotly.express as px
 
+from streamlit_extras.let_it_rain import rain
+
+rain(
+    emoji="⭐",
+    font_size=54,
+    falling_speed=5,
+    animation_length="infinite",
+)
 
 st.header(':green[Text generation by GPT2 model]')
 
@@ -14,7 +22,7 @@ model = GPT2LMHeadModel.from_pretrained(
     output_hidden_states = False,
 )
 
-model.load_state_dict(torch.load('models/model.pt', map_location=torch.device('cpu')))
+model.load_state_dict(torch.load('model.pt', map_location=torch.device('cpu')))
 
 
 length = st.sidebar.slider('**Generated sequence length:**', 8, 256, 15)
@@ -58,4 +66,8 @@ if st.button('**Generate text**'):
                 st.write(textwrap.fill(tokenizer.decode(out_), 100))
                 st.image("pict/wow.png")
             
-            
+
+
+
+
+