Update app.py

app.py

@@ -194,8 +194,8 @@ def update_pdf_gallery_and_extract_text(pdf_files):
     return pdf_files, pdf_text
 
 def get_top_10_cases():
-    result = summarization_pipeline("Show 10 legal case with names and numbers from any State Court of Appeals in the US.", max_length=150, min_length=50, do_sample=False)
-    return result[0]['summary_text']
+    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']
 
 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=150,
+        max_tokens=450,
         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 the top 10 cases")
-                top_10_output = gr.Textbox(label="Top 10 Cases", interactive=False, elem_classes=["scroll-box"])
+                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.click(get_top_10_cases, outputs=top_10_output)
             with gr.Column(scale=2):
                 message = gr.Textbox(label="Case to Argue")