Update app.py

app.py

@@ -13,20 +13,12 @@ def extract_text_from_pdf(pdf_content):
 
 
 def generate_summary(file_content):
-    # Load pre-trained Flan-T5 model
     text_generator = pipeline("text2text-generation", model="google/flan-t5-base")
-
-    # Handle PDF files separately
     if file_content.type == "application/pdf":
         input_text = extract_text_from_pdf(file_content.read())
     else:
-        # Assume text file, directly use the content
         input_text = file_content.read().decode("utf-8")
-
-    # Generate summary
     summary = text_generator(input_text, max_length=1024, num_beams=4)
-
-    # Return results
     return {
         "Extracted Information": input_text,
         "Book Summary": summary[0]["generated_text"],
@@ -36,7 +28,7 @@ def generate_summary(file_content):
 
 iface = gr.Interface(
     fn=generate_summary,
-    inputs=gr.File(),  # No need to specify max_upload_size
+    inputs=gr.File(),
     outputs=[
         gr.Textbox(),
         gr.Textbox(),