Update app.py

app.py

@@ -1,6 +1,6 @@
 import transformers
 import streamlit as st
-from transformers import GPT2LMHeadModel, GPT2Tokenizer
+from transformers import GPT2LMHeadModel, GPT2Tokenizer, GPT2TokenizerFast, GPT2Model
 import numpy as np
 from PIL import Image
 import torch
@@ -16,7 +16,7 @@ max_len = st.slider("Length", 40, 120, 2)
 # tokenizer = GPT2Tokenizer.from_pretrained('sberbank-ai/rugpt3small_based_on_gpt2')
 # #Задаем класс модели (уже в streamlit/tg_bot)
 
-@st.cache
+@st.cache_data
 # def load_gpt():
 #     model_GPT = GPT2LMHeadModel.from_pretrained(
 #      'sberbank-ai/rugpt3small_based_on_gpt2',