Update app.py

app.py

@@ -48,7 +48,7 @@ def proc_submission(
 
     settings = {
         "length_penalty": float(length_penalty),
-        "repetition_penalty": 3.5,#float(repetition_penalty),
+        "repetition_penalty": 3.5,
         "no_repeat_ngram_size": 3,
         "encoder_no_repeat_ngram_size": 4,
         "num_beams": int(num_beams),
@@ -70,13 +70,20 @@ def proc_submission(
     else:
         tr_in = input_text
         msg = None
-
+        
+    if (model_type == "LED" and summary_type == "Detailed"):
+        model = model_led_det
+    elif (model_type == "LED" and summary_type == "TLDR"):
+        model = model_led_tldr
+    elif (model_type == "LongT5" and summary_type == "Detailed"):
+        model = model_det
+    else:
+        model = model_tldr
+        #tokenizer_led_det if (model_type == "LED" and summary_type == "Detailed") else tokenizer_det,
     _summaries = summarize_via_tokenbatches(
         tr_in,
-        model_led_det if (model_type == "LED" and summary_type == "Detailed") else model_det,
-        tokenizer_led_det if (model_type == "LED" and summary_type == "Detailed") else tokenizer_det,
-        model_led_tldr if (model_type == "LED" and summary_type == "TLDR") else model_tldr,
-        tokenizer_led_tldr if (model_type == "LED" and summary_type == "TLDR") else tokenizer_tldr,
+        model = model,
+        tokenizer_led_det if (model_type == "LED") else tokenizer_det,
         batch_length=token_batch_length,
         **settings,
     )