Update app.py

app.py

@@ -86,7 +86,7 @@ def generate_random_poem(prompt, max_length=150, temperature=1.0, top_p=0.9, num
 
 
     
-
+    num_lines = int(num_lines)
     # Ensure that we have exactly 4 lines
     if len(poem_lines) < num_lines:
         while len(poem_lines) < num_lines:
@@ -94,7 +94,7 @@ def generate_random_poem(prompt, max_length=150, temperature=1.0, top_p=0.9, num
             random_line = random.choice(dataset["poem"])
             poem_lines.append(random_line.strip())
 
-    poem_lines = poem_lines[:num_lines]  # Ensure there are only 4 lines
+    poem_lines = poem_lines[:num_lines]
 
     final_poem = "\n".join(poem_lines)