fix line

app.py

@@ -7,32 +7,8 @@ import torch
 import base64
 from PIL import Image
 
-#image = Image.open('../banner.png')
 st.image("https://huggingface.co/spaces/wiwaaw/summary/resolve/main/banner.png")
 
-#st.image(image)
-
-custom_html = """
-<div class="banner">
-    <img src="https://huggingface.co/spaces/wiwaaw/summary/resolve/main/banner.png" alt="Banner Image">
-</div>
-<style>
-    .banner {
-        width: 160%;
-        height: 100%;
-        overflow: hidden;
-    }
-    .banner img {
-        width: 100%;
-        height: 100%;
-        object-fit: cover;
-    }
-</style>
-"""
-# Display the custom HTML
-#st.components.v1.html(custom_html)
-
-
 # Model and tokenizer 
 model_checkpoint = "MBZUAI/LaMini-Flan-T5-783M"
 model_tokenizer = T5Tokenizer.from_pretrained(model_checkpoint)
@@ -56,7 +32,7 @@ def language_model_pipeline(filepath, maxlength):
         model=model,
         tokenizer=model_tokenizer,
         max_length=maxlength, 
-        min_length=70)
+        min_length=32)
     input_text = preprocess_pdf(filepath)
     summary_result = summarization_pipeline(input_text)
     summarized_text = summary_result[0]['summary_text']
@@ -71,9 +47,6 @@ def display_pdf(file):
     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)
 
-# Streamlit code 
-#st.set_page_config(layout="wide")
-
 def main():
     st.title("PDF Summarization App using Language Model")
 
@@ -88,7 +61,6 @@ def main():
                 temp_file.write(uploaded_file.read())
             with col1:
                 st.success("File Uploaded")
-                #pdf_view = display_pdf(filepath)
 
             with col2:
                 summarized_result = language_model_pipeline(filepath, maxlength)