Update app.py

app.py

@@ -10,6 +10,12 @@ def greet(name):
     
     # beam search
     molecules = model.generate(input_ids=sf_input["input_ids"],
+                              attention_mask=sf_input["attention_mask"],
+                              max_length=15,
+                              min_length=5,
+                              num_return_sequences=5,
+                              num_beams=5)
+
     sf_output = [tokenizer.decode(g, skip_special_tokens=True, clean_up_tokenization_spaces=True).replace(" ","") for g in molecules]
     return sf_output