Spaces:

AGIvan
/

title-generation

Sleeping

AGIvan commited on Jul 30, 2024

Commit

f36b7ea

verified ·

1 Parent(s): a7fdfb5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ st.header("Generate candidate titles for articles")
 st_model_load = st.text('Loading title generator model...')
-@st.cache_data(allow_output_mutation=True)
 def load_model():
     print("Loading model...")
     tokenizer = AutoTokenizer.from_pretrained(model_name)

 st_model_load = st.text('Loading title generator model...')
+@st.cache_data()
 def load_model():
     print("Loading model...")
     tokenizer = AutoTokenizer.from_pretrained(model_name)