Update app.py

app.py

@@ -1,35 +1,36 @@
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
+from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 import gradio as gr
 
-# Use lightweight, instruction-tuned model
-model_name = "google/flan-t5-base"
+model_name = "EleutherAI/gpt-neo-1.3B"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
-pipe = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
-# Personas
 personas = {
-    "🟢 Optimist": "Provide a positive and hopeful opinion, mentioning 2 benefits.",
-    "🔴 Pessimist": "Provide a critical opinion, mentioning 2 drawbacks.",
-    "🟡 Neutral": "Provide a balanced perspective, listing pros and cons."
+    "🟢 Optimist": "Give an optimistic argument with 2 benefits and one example.",
+    "🔴 Pessimist": "Give a critical argument with 2 drawbacks and one example.",
+    "🟡 Neutral": "List pros and cons and give a neutral summary."
 }
 
-# Debate function
 def generate_debate(topic):
     results = []
     for label, instruction in personas.items():
-        prompt = f"You are a debater. Topic: '{topic}'. {instruction}"
-        response = pipe(prompt, max_new_tokens=120, temperature=0.7)[0]['generated_text'].strip()
-        results.append(f"### {label}\n{response}")
+        prompt = (
+            f"Debate Topic: {topic}\n"
+            f"Persona: {label}\n"
+            f"{instruction}\n"
+            f"Write 3–4 sentences."
+        )
+        output = pipe(prompt, max_new_tokens=180, temperature=0.7)[0]['generated_text']
+        results.append(f"### {label}\n{output.strip()}")
     return "\n\n".join(results)
 
-# Gradio Interface
 demo = gr.Interface(
     fn=generate_debate,
     inputs=gr.Textbox(label="Debate Topic"),
     outputs=gr.Markdown(),
-    title="🎙️ Multi-Agent Debate Simulator",
-    description="Debate from 3 perspectives (Optimist, Pessimist, Neutral) using FLAN-T5-Base."
+    title="🎙️ Debate Simulator with GPT-Neo",
+    description="Debate with Optimist, Pessimist & Neutral perspectives using GPT-Neo-1.3B."
 )
 
 demo.launch()