Update README.md

README.md

@@ -31,11 +31,11 @@ def generate_story(prompt_text):
     
     output = model.generate(
         input_ids,
-        max_length=200,  # Control output length
-        temperature=0.7,  # Lower value = more deterministic, higher = more creative
-        top_p=0.9,  # Nucleus sampling to avoid repetition
-        top_k=50,  # Consider only top-k likely words
-        repetition_penalty=1.2,  # Reduce repetitive text
+        max_length=200,
+        temperature=0.7,
+        top_p=0.9,
+        top_k=50,
+        repetition_penalty=1.2,
         do_sample=True,
     )