Update app.py

app.py

@@ -101,13 +101,13 @@ if st.button('Generate Molecules'):
     
     # Generate molecules using the model with diverse beam search
     output_ids = model.generate(
-    input_ids=input_ids,
-    max_length=128,
-    do_sample=False,  # Use deterministic beam search
-    num_return_sequences=num_molecules,
-    num_beams=max(num_molecules * 2, 5),  # Ensure enough beams for diversity
-    num_beam_groups=num_molecules,  # Equal to num_return_sequences
-    diversity_penalty=0.5  # Adjust this value as needed
+        input_ids=input_ids,
+        max_length=128,
+        do_sample=False,  # Use deterministic beam search
+        num_return_sequences=num_molecules,
+        num_beams=max(num_molecules * 2, 5),  # Ensure enough beams for diversity
+        num_beam_groups=num_molecules,  # Equal to num_return_sequences
+        diversity_penalty=0.5  # Adjust this value as needed
     )