Update app.py

app.py

@@ -45,12 +45,12 @@ 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
-def generate_random_poem(prompt, max_length=150, temperature=1.0, top_p=0.9, num_lines=4):
+def generate_random_poem(prompt, max_length=180, temperature=1.0, top_p=0.9, num_lines=4):
     # Randomly select a line from the dataset
     random_line = random.choice(dataset["poem"])
 
     # Prepare the input text with the random line selected, and start with a unique phrase to avoid repetition
-    input_text = f"मैया मोरी {random_line} "  # Unique start to force variety
+    input_text = f"{random_line} "  # Unique start to force variety
 
     # Tokenize the input text
     encoding = tokenizer(input_text, return_tensors="pt", padding=True, truncation=True, max_length=max_length)
@@ -93,7 +93,7 @@ interface = gr.Interface(
     fn=generate_random_poem,  # Use function directly without calling it
     inputs=[
         gr.Textbox(label="Enter Prompt", placeholder="Start your Hindi poem..."),
-        gr.Slider(50, 500, step=10, value=100, label="Max Length"),
+        gr.Slider(50, 500, step=10, value=180, label="Max Length"),
         gr.Slider(0.1, 1.5, step=0.1, value=0.7, label="Temperature"),
         gr.Slider(1, 100, step=1, value=50, label="Top-k Sampling"),
         gr.Slider(0.1, 1.0, step=0.05, value=0.95, label="Top-p Sampling"),