update

app.py

@@ -18,6 +18,7 @@ custom_html = """
     }
     .banner img {
         width: 100%;
+        height: 200px;
         object-fit: cover;
     }
 </style>
@@ -43,12 +44,12 @@ def preprocess_pdf(file):
     return final_text
 
 # Language Model pipeline
-def language_model_pipeline(filepath):
+def language_model_pipeline(filepath, maxlength):
     summarization_pipeline = pipeline(
         'summarization',
         model=model,
         tokenizer=model_tokenizer,
-        max_length=500, 
+        max_length=maxlength, 
         min_length=70)
     input_text = preprocess_pdf(filepath)
     summary_result = summarization_pipeline(input_text)
@@ -71,6 +72,7 @@ def main():
     st.title("PDF Summarization App using Language Model")
 
     uploaded_file = st.file_uploader("Upload your PDF file", type=['pdf'])
+    maxlength = st.number_input("Max token", min_value=1, max_value=10, value=5, step=1)
 
     if uploaded_file is not None:
         if st.button("Summarize"):
@@ -83,7 +85,7 @@ def main():
                 pdf_view = display_pdf(filepath)
 
             with col2:
-                summarized_result = language_model_pipeline(filepath)
+                summarized_result = language_model_pipeline(filepath, maxlength)
                 st.info("Summarization Complete")
                 st.success(summarized_result)