haven-jeon commited on
Commit
7c944c3
1 Parent(s): 96484a5
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import streamlit as st
4
  from transformers import GPT2LMHeadModel, PreTrainedTokenizerFast
5
 
6
 
7
- @st.cache
8
  def get_model():
9
  model = GPT2LMHeadModel.from_pretrained('skt/kogpt2-base-v2')
10
  model.eval()
 
4
  from transformers import GPT2LMHeadModel, PreTrainedTokenizerFast
5
 
6
 
7
+ @st.cache(allow_output_mutation=True)
8
  def get_model():
9
  model = GPT2LMHeadModel.from_pretrained('skt/kogpt2-base-v2')
10
  model.eval()