fix line

app.py

@@ -25,6 +25,7 @@ def preprocess_pdf(file):
         final_text = final_text + text.page_content
     return final_text
 
+@st.cache_data
 # Language Model pipeline
 def language_model_pipeline(filepath):
     summarization_pipeline = pipeline(
@@ -38,22 +39,15 @@ def language_model_pipeline(filepath):
     summarized_text = summary_result[0]['summary_text']
     return summarized_text
 
-@st.cache_data
-# Function to display the PDF content
-
-def main():
-    st.title("PDF Summarization using LaMini")
-    uploaded_file = st.file_uploader("Upload your PDF file", type=['pdf'])
-    st.success("File Uploaded")
-    if uploaded_file is not None:
-        if st.button("Summarize"):
-            filepath = uploaded_file.name
-            with open(filepath, "wb") as temp_file:
-                temp_file.write(uploaded_file.read())
+title = st.title("PDF Summarization using LaMini")
+uploaded_file = st.file_uploader("Upload your PDF file", type=['pdf'])
+st.success("File Uploaded")
+if uploaded_file is not None:
+    if st.button("Summarize"):
+        filepath = uploaded_file.name
+        with open(filepath, "wb") as temp_file:
+            temp_file.write(uploaded_file.read())
             
-            summarized_result = language_model_pipeline(filepath)
-            st.info("Summarization Complete")
-            st.success(summarized_result)
-
-if __name__ == "__main__":
-    main()
+        summarized_result = language_model_pipeline(filepath)
+        st.info("Summarization Complete")
+        st.success(summarized_result)