Update app.py

app.py

@@ -137,7 +137,8 @@ class TextGeneration:
 
 
 #@st.cache(hash_funcs={torch.nn.parameter.Parameter: lambda _: None})
-@st.cache(allow_output_mutation=True)
+#@st.cache(allow_output_mutation=True)
+@st.cache(allow_output_mutation=True, hash_funcs={TextGeneration: lambda _: None})
 def load_text_generator():
     text_generator = TextGeneration()
     text_generator.load()