Update app.py

app.py

@@ -107,7 +107,7 @@ if st.button('Generate Molecules'):
         do_sample=True,
         top_k=50,
         num_return_sequences=num_molecules,
-        num_beams=1  # set to 1 when using sampling
+        num_beams = max(num_molecules, 5) # num_beams=1 was taking too long
     )
     
     # Decode generated molecule SMILES