Update app.py

app.py

@@ -82,14 +82,7 @@ def generate_random_poem(prompt, max_length=150, temperature=1.0, top_p=0.9, num
     # Split the generated text into separate lines based on full stops (Hindi poems often end with "।")
     poem_lines = generated_poem.split("।")
 
-    # Ensure we have exactly num_lines (4 by default)
-    if len(poem_lines) < num_lines:
-        while len(poem_lines) < num_lines:
-            # If not enough lines, add more content from the dataset
-            random_line = random.choice(dataset["poem"])
-            poem_lines.append(random_line.strip())
-
-    poem_lines = poem_lines[:num_lines]
+   
 
     final_poem = "\n".join(poem_lines)