Update app.py

app.py

@@ -45,7 +45,6 @@ def generate_poetry(prompt, max_length=100, temperature=0.7, top_k=50, top_p=0.9
     return tokenizer.decode(output[0], skip_special_tokens=True)
 
 # Poetry Generation Function with Random Selection from Dataset and Explicit 4-Line Structure
-# Modify the generate_random_poem function to handle inputs from Gradio
 def generate_random_poem(prompt, max_length=150, temperature=1.0, top_p=0.9, num_lines=4):
     # Randomly select a line from the dataset
     random_line = random.choice(dataset["poem"])
@@ -76,18 +75,14 @@ def generate_random_poem(prompt, max_length=150, temperature=1.0, top_p=0.9, num
         pad_token_id=pad_token_id
     )
 
-    # Decode the output and split into lines using punctuation or newlines
+    # Decode the output and split into lines
     generated_poem = tokenizer.decode(output[0], skip_special_tokens=True)
     generated_poem = generated_poem.strip()
 
-    # Split the generated text into separate lines based on punctuation or line breaks
-    #poem_lines = generated_poem.split("।")  # Hindi poems often end with "।" (full stop)
-    poem_lines  = 4  # Ensure num_lines is an integer
+    # Split the generated text into separate lines based on full stops (Hindi poems often end with "।")
+    poem_lines = generated_poem.split("।")
 
-
-    
-    num_lines = int(num_lines)
-    # Ensure that we have exactly 4 lines
+    # 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
@@ -117,4 +112,3 @@ interface = gr.Interface(
 
 # 4️⃣ Run the Gradio App
 interface.launch(share=True)
-