Update app.py

app.py

@@ -2,11 +2,10 @@ import gradio as gr
 from transformers import pipeline
 
 # Load model (You can replace this with any model like GPT-2 or your custom one)
-generator = pipeline("text-generation", model="aniketmaurya/product-description-generator")
-
+generator = pipeline("text-generation", model="gpt2")
 
 def generate_description(title, tags):
-    prompt = f"Generate an SEO-friendly product description for a product titled '{title}' with tags: {tags}."
+    prompt = f"Write a clear, professional, and SEO-optimized product description for an e-commerce listing. The product title is: '{title}'. The relevant tags or features include: {tags}. Mention key benefits, usage, and appeal to buyers."
     result = generator(prompt, max_length=100, num_return_sequences=1)
     return result[0]['generated_text']