cpu

agent.py

@@ -29,7 +29,7 @@ class NewsReporterAgent:
         print("--- 🚀 INITIALIZING MODEL (this may take a moment) ---")
         
         # Define the Hugging Face Hub model ID
-        model_id = "google/gemma-3n-E4B-it" 
+        model_id = "google/gemma-3n-E2B-it" # google/gemma-3n-E4B-it # 4b taking too long
         
         self.device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
         print(f"  > Using device: {self.device}")

requirements.txt

@@ -1,6 +1,6 @@
 # Core ML/AI Libraries
 torch
-transformers
+transformers==4.54.1
 langchain-core
 sentencepiece
 typing_extensions