Update app.py

update h-paramters

app.py

@@ -24,13 +24,13 @@ def load_model(model_name="BioGPT"):
   return tokenizer, model
 
 
-def get_beam_output(sentence, selected_model, min_len,max_len, n_beams):
+def get_beam_output(sentence, selected_model, min_len=100,max_len=512, n_beams=1):
     tokenizer, model = load_model(selected_model)
     inputs = tokenizer(sentence, return_tensors="pt")
     with torch.no_grad():
         beam_output = model.generate(**inputs,
-                                    min_length=100,
-                                    max_length=1024,
+                                    min_length=min_len,
+                                    max_length=max_len,
                                     num_beams=n_beams,
                                     early_stopping=True
                                     )