Update app.py

app.py

@@ -7,11 +7,13 @@ from dotenv import load_dotenv
 # Load environment variables
 load_dotenv()
 
-# Configuration
+# Configuration - Changed to SDXL for image generation
 HF_TOKEN = os.getenv("HF_TOKEN")
 REDDIT_API = "https://www.reddit.com/r/trending/top.json?limit=5"
 DEFAULT_TRENDS = ["Celebrity gossip", "Tech news", "Movie drama", "Gaming leaks"]
 STYLES = ["Realistic", "Cartoon", "Anime", "3D Render"]
+IMAGE_MODEL = "stabilityai/stable-diffusion-xl-base-1.0"  # Verified working model
+TEXT_MODEL = "deepseek-ai/Janus-Pro-7B"  # For text processing
 
 # Initialize clients
 client = InferenceClient(token=HF_TOKEN)
@@ -25,16 +27,19 @@ def get_trends():
         return DEFAULT_TRENDS
 
 def generate_meme(prompt: str, style: str):
-    """Generate meme using DeepSeek Janus-Pro"""
+    """Generate meme using SDXL for images and Janus-Pro for text refinement"""
     try:
-        enhanced_prompt = (
-            f"Trending meme template ({style} style): {prompt}. "
-            "Viral social media format, high contrast, funny text overlay"
+        # First enhance text with Janus-Pro
+        enhanced_prompt = client.text_generation(
+            model=TEXT_MODEL,
+            prompt=f"Improve this meme caption about {prompt} in {style} style:",
+            max_new_tokens=100
         )
         
+        # Generate image with SDXL
         image = client.text_to_image(
-            model="deepseek-ai/Janus-Pro-7B",
-            prompt=enhanced_prompt,
+            model=IMAGE_MODEL,
+            prompt=f"Trending meme template ({style} style): {enhanced_prompt}",
             negative_prompt="low quality, text, watermark",
             height=512,
             width=512
@@ -48,16 +53,16 @@ def generate_meme(prompt: str, style: str):
 with gr.Blocks(title="🚀 Viral Meme Generator", css="static/style.css") as demo:
     gr.Markdown("# <center>🔥 Create Viral Memes in Seconds</center>")
     
-    with gr.Row():  # Removed invalid 'variant' parameter
+    with gr.Row():
         with gr.Column(scale=3):
-            gr.HTML('<img src="file/static/assets/logo.png" style="height: 200px">')  # Fixed image sizing
+            gr.HTML('<img src="file/static/assets/logo.png" style="height: 200px">')
             trend_select = gr.Dropdown(get_trends(), label="Trending Topics")
             style_select = gr.Dropdown(STYLES, label="Visual Style", value="Realistic")
             text_input = gr.Textbox(label="Your Message", placeholder="Add funny text...")
             generate_btn = gr.Button("Generate Now", variant="primary")
             
         with gr.Column(scale=2):
-            output_img = gr.Image(label="Your Meme", height=512, width=512)  # Fixed height/width
+            output_img = gr.Image(label="Your Meme", height=512, width=512)
             gr.HTML("""
             <div class="monetization">
                 <script type='text/javascript' src='https://storage.ko-fi.com/cdn/widget/Widget_2.js'></script>