Update app.py

app.py

@@ -71,22 +71,22 @@ def proc_submission(
         tr_in = input_text
         msg = None
         
-    if (model_type == "LED" and summary_type == "Detailed"):
-        m = model_led_det
-    elif (model_type == "LED" and summary_type == "TLDR"):
-        m = model_led_tldr
-    elif (model_type == "LongT5" and summary_type == "Detailed"):
-        m = model_det
-    elif (model_type == "LongT5" and summary_type == "TLDR"):
-        m = model_tldr
-        #tokenizer_led_det if (model_type == "LED" and summary_type == "Detailed") else tokenizer_det,
-    _summaries = summarize_via_tokenbatches(
-        tr_in,
-        tokenizer_led_det if (model_type == "LED") else tokenizer_det,
-        model = m,
-        batch_length=token_batch_length,
-        **settings,
-    )
+    if (model_type == "LED"):
+        _summaries = summarize_via_tokenbatches(
+            tr_in,
+            model_led_tldr if (summary_type == "TLDR") else model_led_det,
+            tokenizer_led_tldr if (summary_type == "TLDR") else tokenizer_led_det,
+            batch_length=token_batch_length,
+            **settings,
+        )
+    if (model_type == "LongT5"):
+        _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)}"