Update app.py

app.py

@@ -4,7 +4,6 @@ import torch
 import random
 from datasets import load_dataset
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
-import torch
 
 # Load dataset
 dataset = load_dataset("rahul7star/hindi-poetry")["train"]
@@ -14,14 +13,6 @@ model_name = "rahul7star/hindi_poetry_language_model"
 tokenizer = GPT2Tokenizer.from_pretrained(model_name)
 model = GPT2LMHeadModel.from_pretrained(model_name)
 
-# 1️⃣ Load Model & Tokenizer from Hugging Face Hub
-# model_name = "rahul7star/hindi_poetry_language_model"
-# tokenizer = AutoTokenizer.from_pretrained(model_name)
-# model = AutoModelForCausalLM.from_pretrained(model_name)
-
-# Ensure model uses the correct pad token
-#model.config.pad_token_id = tokenizer.pad_token_id
-
 # 2️⃣ Function to Generate Hindi Poetry
 def generate_poetry_base(prompt, max_length=100, temperature=0.7, top_k=50, top_p=0.95):
     input_ids = tokenizer(prompt, return_tensors="pt").input_ids
@@ -41,7 +32,6 @@ def generate_poetry(prompt, max_length=100, temperature=0.7, top_k=50, top_p=0.9
     with torch.no_grad():
         output = model.generate(
             input_ids,
-           
             max_length=max_length,
             temperature=temperature,  # Increased randomness
             top_p=top_p,
@@ -54,11 +44,8 @@ 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(num_lines=4, max_length=150, temperature=1.0, top_p=0.9):
+def generate_random_poem(prompt, num_lines=4, max_length=150, temperature=1.0, top_p=0.9):
     # Randomly select a line from the dataset
     random_line = random.choice(dataset["poem"])
 
@@ -111,16 +98,9 @@ def generate_random_poem(num_lines=4, max_length=150, temperature=1.0, top_p=0.9
 
     return final_poem
 
-
-
-
-
-
-
 # 3️⃣ Gradio Interface
 interface = gr.Interface(
-    #fn=generate_poetry,
-    fn=generate_random_poem(num_lines=4),
+    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"),
@@ -130,7 +110,7 @@ interface = gr.Interface(
     ],
     outputs=gr.Textbox(label="Generated Hindi Poem"),
     title="Hindi Poetry Generator ✨",
-    description="Generate beautiful Hindi poetry Just enter a prompt and adjust parameters  Exp - मैया मोरी",
+    description="Generate beautiful Hindi poetry. Just enter a prompt and adjust parameters. Example: 'मैया मोरी'",
 )
 
 # 4️⃣ Run the Gradio App