Update app.py

app.py

@@ -20,12 +20,11 @@ logging.basicConfig(
 
 def proc_submission(
     input_text: str,
-    model_type: str,
-    summary_type: str,
+    model_type: str = 'LongT5',
+    summary_type: str = 'TLDR',
     num_beams,
     token_batch_length,
     length_penalty,
-    #repetition_penalty,
     #no_repeat_ngram_size: int = 3,
     max_input_length: int = 768,
 ):
@@ -71,7 +70,7 @@ def proc_submission(
         tr_in = input_text
         msg = None
         
-    if (model_type == "LED"):
+    if model_type == "LED":
         _summaries = summarize_via_tokenbatches(
             tr_in,
             model_led_tldr if (summary_type == "TLDR") else model_led_det,
@@ -79,7 +78,7 @@ def proc_submission(
             batch_length=token_batch_length,
             **settings,
         )
-    elif (model_type == "LongT5"):
+    elif model_type == "LongT5":
         _summaries = summarize_via_tokenbatches(
             tr_in,
             model_tldr if (summary_type == "TLDR") else model_det,
@@ -87,13 +86,13 @@ def proc_submission(
             batch_length=token_batch_length,
             **settings,
         )
-    _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,
-    )
+    #_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,
+    #)
     sum_text = [f"Section {i}: " + s["summary"][0] for i, s in enumerate(_summaries)]
     compression_rate = [
         f" - Section {i}: {round(s['compression_rate'],3)}"