Update app.py

app.py

@@ -4,34 +4,32 @@ from transformers import pipeline
 import gradio as gr
 
 # Load FLAN-T5-LARGE pipeline
-generator = pipeline("text2text-generation", model="google/flan-t5-large")
+generator = pipeline("text-generation", model="gpt2-large")
 
 def generate_brand_identity(startup_idea):
-    # Make each prompt practical and targeted
-
     # Brand Names
-    prompt_name = f"Generate 3 unique and creative brand names for a startup that offers: {startup_idea}. The names should be catchy and relevant. Separate with commas."
+    prompt_name = f"Give three short, creative, and unique brand name ideas for a startup that is {startup_idea}. Avoid repetition and ensure they are realistic for a business."
     brand_names = generator(prompt_name, max_length=60, num_return_sequences=1)[0]["generated_text"]
 
     # Tagline
-    tagline_prompt = f"Write a 1-line catchy tagline for a startup that offers: {startup_idea}. Focus on value, not just flair."
+    tagline_prompt = f"Generate a catchy and original tagline for a startup focused on {startup_idea}. Keep it under 10 words."
     tagline = generator(tagline_prompt, max_length=30, num_return_sequences=1)[0]['generated_text']
 
-    # Brand Personality Description
-    personality_prompt = f"Describe the brand personality for a startup that offers: {startup_idea}. Use descriptive language."
-    personality = generator(personality_prompt, max_length=50, num_return_sequences=1)[0]['generated_text']
+    # Brand Personality
+    personality_prompt = f"Describe the brand personality of a startup doing {startup_idea} in one powerful sentence. Use descriptive adjectives, not vague words."
+    personality = generator(personality_prompt, max_length=40, num_return_sequences=1)[0]['generated_text']
 
-    # Suggested Brand Colours
-    colours_prompt = f"Suggest two brand colour themes (primary and secondary) with HEX codes for a startup that offers: {startup_idea}." 
+    # Brand Colours
+    colours_prompt = f"Suggest a primary and secondary brand colour with HEX codes for a startup about {startup_idea}. Format: Color Name - HEX"
     colours = generator(colours_prompt, max_length=40, num_return_sequences=1)[0]['generated_text']
 
-    # Target Audience Description
-    audience_prompt = f"Identify the most likely target audience for a startup that offers: {startup_idea}. Include age, profession, or lifestyle."
+    # Target Audience
+    audience_prompt = f"Describe the ideal target audience (age, profession, need) for a startup doing {startup_idea}."
     audience = generator(audience_prompt, max_length=50, num_return_sequences=1)[0]['generated_text']
 
-    # Slogan Options
-    slogan_prompt = f"Generate 2 short, memorable slogans for a startup that offers: {startup_idea}. Keep them unique and practical."
-    slogan = generator(slogan_prompt, max_length=40, num_return_sequences=1)[0]['generated_text']
+    # Slogan
+    slogan_prompt = f"Suggest a memorable and short slogan (max 7 words) for a startup about {startup_idea}. Make it inspirational or witty."
+    slogan = generator(slogan_prompt, max_length=20, num_return_sequences=1)[0]['generated_text']
 
     return (
         brand_names.strip(),