Update app.py

app.py

@@ -95,14 +95,20 @@ def proc_submission(
             #tokenizer_led_det if (model_type == "LED") else tokenizer_det,
             #batch_length=token_batch_length,
             #**settings_det,
-        #)    
+        #)
+    if summary_type == 'TLDR':
+        settings = settings_tldr
+    elif summary types == 'Detailed':
+        settings = settings_det
+           
     _summaries = summarize_via_tokenbatches(
         tr_in,
         model_tldr if (summary_type == "TLDR") else model_det,
         tokenizer_tldr if (summary_type == "TLDR") else tokenizer_det,
         batch_length=token_batch_length,
-        **settings_tldr,# if (summary_type == "TLDR") else **settings_det,
+        **settings,
     )
+    
     sum_text = [f"Section {i}: " + s["summary"][0] for i, s in enumerate(_summaries)]
     compression_rate = [
         f" - Section {i}: {round(s['compression_rate'],3)}"
@@ -192,7 +198,7 @@ if __name__ == "__main__":
             )
             with gr.Row():
                 summary_type = gr.Radio(
-                    choices=["TLDR", "Detailed"], label="Summary Type", value="Detailed"
+                    choices=["TLDR", "Detailed"], label="Summary Type", value="TLDR"
                 )
                 model_type = gr.Radio(
                     choices=["LongT5", "LED"], label="Model Architecture", value="LongT5"
@@ -216,22 +222,9 @@ if __name__ == "__main__":
                 token_batch_length = gr.Radio(
                     choices=[512, 768, 1024],
                     label="token batch length",
-                    value=512,
+                    value=1024,
                 )
 
-            #with gr.Row():
-                #repetition_penalty = gr.inputs.Slider(
-                    #minimum=1.0,
-                    #maximum=5.0,
-                    #label="repetition penalty",
-                    #default=3.5,
-                    #step=0.1,
-                #)
-                #no_repeat_ngram_size = gr.Radio(
-                    #choices=[2, 3, 4],
-                    #label="no repeat ngram size",
-                    #value=3,
-               # )
             with gr.Row():
                 example_name = gr.Dropdown(
                     list(name_to_path.keys()),