Update app.py

app.py

@@ -14,9 +14,8 @@ import requests
 import glob
 
 # Model for OpenRouter
-# Using paid llama-3.2-3b-instruct since free tier models have unreliable rate limits
 MODELS = [
-    "meta-llama/llama-3.2-3b-instruct",  # 3B - Reliable, fast, and very cheap ($0.04/$0.04 per 1M tokens)
+    "google/gemini-2.5-flash-lite",  
 ]
 
 
@@ -778,7 +777,7 @@ Your improved prompt here
 
     config = {
         "llm": {
-            "primary_model": "meta-llama/llama-3.1-8b-instruct",  # Use STRONGER model for prompt generation
+            "primary_model": "google/gemini-2.5-flash",  # Use STRONGER model for prompt generation
             "api_base": "https://openrouter.ai/api/v1",  # Use OpenRouter endpoint
             "temperature": 1.2,  # Even higher temperature for more creative variations
         },
@@ -1007,7 +1006,7 @@ def optimize_prompt(initial_prompt: str, dataset_name: str, dataset_split: str,
 ### Summary
 - **Dataset**: {dataset_name} ({dataset_split} split)
 - **Evaluation Model**: {model}
-- **Evolution Model**: meta-llama/llama-3.1-8b-instruct (larger model for better prompt generation)
+- **Evolution Model**: google/gemini-2.5-flash (larger model for better prompt generation)
 - **Initial Eval**: 50 samples
 - **Final Eval**: 50 samples (same samples for fair comparison)
 - **Evolution**: 50 samples per variant (SAME samples as initial/final!)
@@ -1045,7 +1044,7 @@ with gr.Blocks(title="OpenEvolve Prompt Optimizer", theme=gr.themes.Soft()) as d
 
     **Usage**: Enter initial prompt with `{input}` placeholder → Click optimize → Compare results
 
-    **Model**: `meta-llama/llama-3.2-3b-instruct` (~$0.04 per 1M tokens)
+    **Model**: `google/gemini-2.5-flash-lite` 
     """)
 
     with gr.Row():
@@ -1115,7 +1114,7 @@ with gr.Blocks(title="OpenEvolve Prompt Optimizer", theme=gr.themes.Soft()) as d
         """Wrapper to use fixed model instead of dropdown"""
         return optimize_prompt(
             initial_prompt, dataset_name, dataset_split,
-            MODELS[0],  # Use fixed llama-3.2-3b model
+            MODELS[0],  
             input_field, target_field, progress
         )