Updated

main.py

@@ -1,5 +1,5 @@
 # ==============================================================
-# Tech Disciples AI Backend — Free-Response Edition
+# Tech Disciples AI Backend — Free-Response Edition (Flan-T5 Fixed)
 # ==============================================================
 
 import os
@@ -8,7 +8,7 @@ import torch
 from fastapi import FastAPI, Header, HTTPException
 from fastapi.responses import JSONResponse
 from pydantic import BaseModel
-from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 
 # --------------------------------------------------------------
 # Logging Setup
@@ -57,7 +57,7 @@ class ChatRequest(BaseModel):
     query: str
 
 # --------------------------------------------------------------
-# System Prompt (always prepended automatically)
+# System Prompt
 # --------------------------------------------------------------
 SYSTEM_TEMPLATE = """You are Tech Disciples AI — a spiritually aware, intelligent, and kind conversational assistant.
 You respond clearly and truthfully, offering thoughtful, biblically grounded, and intelligent answers.
@@ -77,7 +77,7 @@ chat_pipe = None
 try:
     logger.info(f"🚀 Loading model: {MODEL_NAME}")
     tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, token=HF_TOKEN if HF_TOKEN else None)
-    model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, token=HF_TOKEN if HF_TOKEN else None)
+    model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME, token=HF_TOKEN if HF_TOKEN else None)
 
     chat_pipe = pipeline(
         "text2text-generation",
@@ -103,7 +103,7 @@ def generate_reply(pipe, user_input: str) -> str:
         result = pipe(
             prompt,
             do_sample=True,
-            temperature=0.3,     # Balanced creativity & logic
+            temperature=0.3,
             top_p=0.9,
             repetition_penalty=1.05
         )
@@ -133,4 +133,3 @@ async def ai_chat(req: ChatRequest, authorization: str | None = Header(None)):
 # ==============================================================
 # END OF FILE
 # ==============================================================
-