Update src/app/main_agent.py

src/app/main_agent.py

@@ -57,8 +57,10 @@ import torch
 from transformers import pipeline
 import os
 
+model_id = "google/gemma-3-4b-it"
+
 # Load the Gemma 3 model pipeline once
-pipe = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1", use_auth_token=os.getenv("HF_TOKEN"))
+pipe = pipeline("text-generation", model=model_id, use_auth_token=os.getenv("HF_TOKEN"))
 
 def create_agent(accent_tool_obj) -> tuple[Runnable, Runnable]:
     accent_tool = Tool(