Update app.py

app.py

@@ -192,7 +192,7 @@ def predict(
 def proc_submission(
     input_text: str,
     model_name: str,
-    num_beams: int,
+    num_beams: int = 3,
     token_batch_length: int,
     length_penalty: float,
     repetition_penalty: float,
@@ -228,7 +228,7 @@ def proc_submission(
         "repetition_penalty": float(repetition_penalty),
         "no_repeat_ngram_size": int(no_repeat_ngram_size),
         "encoder_no_repeat_ngram_size": 4,
-        "num_beams": 3,
+        "num_beams": int(num_beams),
         "min_length": 4,
         "max_length": int(token_batch_length // 4),
         "early_stopping": True,