Update app.py

app.py

@@ -24,7 +24,7 @@ def proc_submission(
     num_beams,
     token_batch_length,
     length_penalty,
-    max_input_length: int = 768,
+    max_input_length: int = 1000,
 ):
     """
     proc_submission - a helper function for the gradio module to process submissions
@@ -60,7 +60,7 @@ def proc_submission(
         "encoder_no_repeat_ngram_size": 4,
         "num_beams": int(num_beams),
         "min_length": 100,
-        "max_length": 512#int(token_batch_length // 4) if (token_batch_length <500) else 512,
+        "max_length": 512,#int(token_batch_length // 4) if (token_batch_length <500) else 512,
         "early_stopping": True,
     }
     st = time.perf_counter()
@@ -201,7 +201,7 @@ if __name__ == "__main__":
                     choices=["LongT5", "LED"], label="Model Architecture", value="LongT5"
                 )
                 num_beams = gr.Radio(
-                    choices=[2, 3, 4, 5, 6],
+                    choices=[2, 3, 4],
                     label="Beam Search: # of Beams",
                     value=2,
                 )