Update app.py

app.py

@@ -4,5 +4,18 @@ import streamlit
 
 cp_aug = 'minnehwg/finetune-newwiki-summarization-ver-augmented2'
 
+@st.cache_resource
+def get_model():
+    checkpoint = cp_aug 
+    tokenizer, model = load_model(checkpoint)
+    return tokenizer, model
 
-tokenizer, model_aug = load_model(cp_aug)
+
+st.title("Chào mừng đến với hệ thống tóm tắt của Minne >.< ")
+tokenizer, model = get_model()
+input_text = st.text_area("Enter your URL:")
+    
+if st.button("Generate"):
+    output_text = pipeline(input_text)
+
+    st.text_area("Generated Text:", value=output_text, height=200)