Update app.py

app.py

@@ -22,7 +22,7 @@ checkpoint = "MBZUAI/LaMini-Flan-T5-248M"
 #model and tokenizer loading
 
 tokenizer = T5Tokenizer.from_pretrained(checkpoint)
-base_model = T5ForConditionalGeneration.from_pretrained(checkpoint, device_map='auto', torch_dtype=torch.float32, offload_folder='D:/project/offload')
+base_model = T5ForConditionalGeneration.from_pretrained(checkpoint, device_map='auto', torch_dtype=torch.float32)
 
 #file loader and preprocessing
 def file_preprocessing(file):
@@ -48,7 +48,7 @@ def llm_pipeline(filepath):
     result = pipe_sum(input_text)
     result = result[0]['summary_text']
     return result
-
+"""
 @st.cache_data
 #function to display the PDF of a given file 
 def displayPDF(file):
@@ -61,6 +61,7 @@ def displayPDF(file):
 
     # Displaying File
     st.markdown(pdf_display, unsafe_allow_html=True)
+    """
 
 #streamlit code 
 st.set_page_config(layout="wide")
@@ -79,10 +80,10 @@ def main():
                 temp_file.flush()  # Ensure contents are written to disk
                 filepath = temp_file.name
             
-            with col1:
+         """   with col1:
                 st.info("Uploaded File")
                 pdf_view = displayPDF(filepath)
-
+"""
             with col2:
                 summary = llm_pipeline(filepath)
                 st.info("Summarization Complete")