Update app.py

app.py

@@ -1,14 +1,13 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load the powerful open-source model
+# Load GPT-Neo model, optimized for CPU usage
 generator = pipeline(
     "text-generation",
-    model="mistralai/Mistral-7B-Instruct-v0.1",
-    device_map="auto"  # If running locally with GPU; otherwise remove
+    model="EleutherAI/gpt-neo-1.3B"  # Runs fine on CPU Basic tier
 )
 
-# Define max length recommendations
+# Recommended length per content type
 max_lengths = {
     "social media post": 280,
     "email newsletter": 800,
@@ -18,15 +17,15 @@ max_lengths = {
 
 def generate_marketing_text(prompt, content_type, _, temperature=0.7):
     """
-    Generate marketing text using Mistral 7B based on topic and content type.
+    Generate marketing text using GPT-Neo based on topic and content type.
     """
-    # Clean and enhance prompt
-    enhanced_prompt = f"Write a {content_type} for this product: {prompt}\nMake it engaging, professional, and persuasive."
+    # Smart prompt engineering for better output
+    enhanced_prompt = f"Write a {content_type} for the following product: {prompt}\nMake it persuasive, professional, and engaging."
 
-    # Determine proper max_length for the type
+    # Choose max_length based on content type
     max_length = max_lengths.get(content_type, 400)
 
-    # Generate the text
+    # Generate text using the model
     result = generator(
         enhanced_prompt,
         max_length=max_length,
@@ -37,7 +36,7 @@ def generate_marketing_text(prompt, content_type, _, temperature=0.7):
 
     return result[0]['generated_text']
 
-# Gradio UI
+# Gradio UI definition
 demo = gr.Interface(
     fn=generate_marketing_text,
     inputs=[
@@ -47,12 +46,13 @@ demo = gr.Interface(
             label="Content Type",
             value="social media post"
         ),
-        gr.Slider(minimum=50, maximum=800, value=280, step=10, label="(Disabled) Max Length", interactive=False),
+        gr.Slider(minimum=50, maximum=800, value=280, step=10, label="(Auto-set) Max Length", interactive=False),
         gr.Slider(minimum=0.1, maximum=1.0, value=0.7, step=0.1, label="Creativity (Temperature)")
     ],
     outputs=gr.Textbox(lines=10, label="Generated Marketing Content"),
     title="AdGenAI - Marketing Content Generator",
-    description="Enter a topic and select a content type to generate persuasive, high-quality marketing content using Mistral 7B."
+    description="Free-tier friendly version using GPT-Neo 1.3B. Enter a topic and select the type of content you want to generate."
 )
 
+# Launch the app
 demo.launch()