Spaces:
Build error
Build error
haven-jeon
commited on
Commit
•
7c944c3
1
Parent(s):
96484a5
fix cache
Browse files
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()
|