Update app.py

app.py

@@ -4,10 +4,11 @@ from transformers import pipeline
 
 # Use a pipeline as a high-level helper
 device = 0 if torch.cuda.is_available() else -1
-text_summary = pipeline("summarization", model="Falconsai/text_summarization",device=device,torch_dtype=torch.bfloat16)
+text_summary = pipeline("summarization", model="Falconsai/text_summarization", device=device, torch_dtype=torch.bfloat16)
 
 def summary(input):
-    output = text_summary(input)
+    # Adjust max_length and min_length for better summarization
+    output = text_summary(input, max_length=200, min_length=50, truncation=True)
     return output[0]['summary_text']
 
 gr.close_all()
@@ -22,3 +23,4 @@ demo = gr.Interface(
 )
 
 demo.launch()
+