Update app.py

app.py

@@ -5,12 +5,6 @@ from transformers import pipeline
 # Initialize Model
 get_completion = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", device=0)
 
-
-# @spaces.GPU(duration=120)
-# def summarize(input):
-#     output = get_completion(input)
-#     return output[0]['summary_text']
-
 @spaces.GPU(duration=120)
 def summarize(input: str) -> str:
     """
@@ -51,7 +45,7 @@ with demo:
   gr.Markdown(title)
   gr.Markdown(description)
   interface = gr.Interface(fn=summarize, 
-                    inputs=[gr.Textbox(label="Text to Summarize", lines=6)], 
-                    outputs=[gr.Textbox(label="Result", lines=3)]
+                    inputs=[gr.Textbox(label="Text to Summarize", lines=15)], 
+                    outputs=[gr.Textbox(label="Result", lines=7)]
                    )
 demo.launch()