Update app.py

app.py

@@ -9,7 +9,6 @@ import pandas as pd
 
 # Small, free chat models that run on CPU in a basic Space (pick one if you like)
 DEFAULT_MODELS = [
-    "google/gemma-2-2b-it",
     "TinyLlama/TinyLlama-1.1B-Chat-v1.0",
     "Qwen/Qwen2.5-1.5B-Instruct",
 ]
@@ -176,6 +175,7 @@ with gr.Blocks(title="Multi-Prompt Chat (System Prompt Control)") as demo:
                 wrap=True,
                 interactive=False,
                 row_count=(0, "dynamic"),
+                type="pandas",
             )
             out_file = gr.File(label="CSV file", visible=False)
 
@@ -193,14 +193,9 @@ with gr.Blocks(title="Multi-Prompt Chat (System Prompt Control)") as demo:
         return df
 
     def _download(df):
-        # Gradio passes a dict-like table; normalise to DataFrame
-        if isinstance(df, list):
-            df = pd.DataFrame(df, columns=["user_prompt", "response", "tokens_out"])
-        else:
-            df = pd.DataFrame(df)
         path = to_csv(df)
         return gr.File.update(value=path, visible=True)
-
+    
     run_btn.click(
         _generate,
         inputs=[model_id, system_prompt, prompts_multiline, max_new_tokens, temperature, top_p, top_k, repetition_penalty],