fix line

app.py

@@ -26,12 +26,12 @@ def preprocess_pdf(file):
     return final_text
 
 # Language Model pipeline
-def language_model_pipeline(filepath, maxlength):
+def language_model_pipeline(filepath):
     summarization_pipeline = pipeline(
         'summarization',
         model=model,
         tokenizer=model_tokenizer,
-        max_length=maxlength, 
+        max_length=500, 
         min_length=32)
     input_text = preprocess_pdf(filepath)
     summary_result = summarization_pipeline(input_text)
@@ -40,31 +40,20 @@ def language_model_pipeline(filepath, maxlength):
 
 @st.cache_data
 # Function to display the PDF content
-def display_pdf(file):
-    with open(file, "rb") as f:
-        base64_pdf = base64.b64encode(f.read()).decode('utf-8')
-
-    pdf_display = f'<object data="data:application/pdf;base64,{base64_pdf}" width="100%" height="600" type="application/pdf"></object>'
-    st.markdown(pdf_display, unsafe_allow_html=True)
 
 def main():
     st.title("PDF Summarization App using Language Model")
     uploaded_file = st.file_uploader("Upload your PDF file", type=['pdf'])
-    maxlength = st.slider(label="Token Length", min_value=32, max_value=1024, step=32, value=1024)
-
+    st.success("File Uploaded")
     if uploaded_file is not None:
         if st.button("Summarize"):
-            col1, col2 = st.columns(2)
             filepath = uploaded_file.name
             with open(filepath, "wb") as temp_file:
                 temp_file.write(uploaded_file.read())
-            with col1:
-                st.success("File Uploaded")
-
-            with col2:
-                summarized_result = language_model_pipeline(filepath, maxlength)
-                st.info("Summarization Complete")
-                st.success(summarized_result)
+            
+            summarized_result = language_model_pipeline(filepath)
+            st.info("Summarization Complete")
+            st.success(summarized_result)
 
 if __name__ == "__main__":
     main()