Update app.py

app.py

@@ -11,7 +11,7 @@ from transformers import pipeline
 from datasets import load_dataset
 import fitz  # PyMuPDF
 
-# Loading datasets
+# Load dataset
 dataset = load_dataset("ibunescu/qa_legal_dataset_train")
 
 # Different pipelines for different tasks
@@ -82,7 +82,7 @@ def determine_winner(outcome):
     else:
         return "No clear winner"
 
-# Custom CSS for white background and black text for input and output boxes
+# Custom CSS for a clean layout
 custom_css = """
 body {
     background-color: #ffffff;
@@ -194,8 +194,8 @@ def update_pdf_gallery_and_extract_text(pdf_files):
     return pdf_files, pdf_text
 
 def get_top_10_cases():
-    result = mask_filling_pipeline("Show 5 legal case with names and numbers from any State Court of Appeals in the US.", max_length=450, min_length=50, do_sample=False)
-    return result[0]['text']
+    summary = summarization_pipeline("Show 10 legal cases with names and numbers from any State Court of Appeals in the US.", max_length=150, min_length=50, do_sample=False)
+    return summary[0]['summary_text']
 
 def add_message(history, message):
     for x in message["files"]:
@@ -215,7 +215,7 @@ def bot(history):
     response = ""
     for message in client.chat_completion(
         messages,
-        max_tokens=450,
+        max_tokens=150,
         stream=True,
         temperature=0.6,
         top_p=0.95,
@@ -250,8 +250,8 @@ with gr.Blocks(css=custom_css) as demo:
     with gr.Tab("Argument Evaluation"):
         with gr.Row():
             with gr.Column(scale=1):
-                top_10_btn = gr.Button("Give me 5 cases")
-                top_10_output = gr.Textbox(label="Show 5 Cases", interactive=False, elem_classes=["scroll-box"])
+                top_10_btn = gr.Button("Give me the top 10 cases")
+                top_10_output = gr.Textbox(label="Top 10 Cases", interactive=False, elem_classes=["scroll-box"])
                 top_10_btn.click(get_top_10_cases, outputs=top_10_output)
             with gr.Column(scale=2):
                 message = gr.Textbox(label="Case to Argue")