Update app.py

app.py

@@ -1,4 +1,4 @@
-# Import libraries
+# Imports
 import gradio as gr
 from transformers import AutoModel, AutoProcessor
 import torch
@@ -13,7 +13,7 @@ model_name = "facebook/bart-large-cnn"
 # Create a summarization pipeline with Facebook Bart
 summarizer = pipeline("summarization", model=model_name)
 
-# Define a function that takes a text and returns a summary
+# Define a function that takes input and returns summary
 def summarize(input_text):
   summary = summarizer(input_text, max_length=100, min_length=10, do_sample=False)
   return summary[0]["summary_text"]
@@ -21,8 +21,8 @@ def summarize(input_text):
 # Create a Gradio interface
 interface = gr.Interface(
   fn=summarize, # the function to wrap
-  inputs=gr.Textbox(lines=10, label="Input Text"), # the input component
-  outputs=gr.Textbox(label="Summary") # the output component
+  inputs=gr.Textbox(lines=10, label="Input Text"),
+  outputs=gr.Textbox(label="Summary")
 )
 
 # Launch the interface