Update src/app/main_agent.py

src/app/main_agent.py

@@ -57,10 +57,11 @@ import torch
 from transformers import pipeline
 import os
 
+model_name = 'meta-llama/Llama-3.1-8B' #"google/gemma-3-4b-it"
 # Load the Gemma 3 model pipeline once
 gemma_pipeline = pipeline(
     task="text-generation",
-    model="google/gemma-3-4b-it",  # or your preferred Gemma 3 model
+    model= model_name,  # or your preferred Gemma 3 model
     device=0,  # set -1 for CPU, 0 or other for GPU
     torch_dtype=torch.bfloat16,
     use_auth_token=os.getenv("HF_TOKEN")