Update app.py

app.py

@@ -42,6 +42,17 @@ class BasicAgent:
     
     def __init__(self):
         print("SimpleCodeAgent initialized.")
+
+        # Check GPU availability
+        print(f"🔍 GPU Check:")
+        print(f"   - CUDA available: {torch.cuda.is_available()}")
+        if torch.cuda.is_available():
+            print(f"   - CUDA device count: {torch.cuda.device_count()}")
+            print(f"   - Current device: {torch.cuda.current_device()}")
+            print(f"   - Device name: {torch.cuda.get_device_name()}")
+            print(f"   - Device memory: {torch.cuda.get_device_properties(0).total_memory / 1e9:.1f} GB")
+        else:
+            print("   - No CUDA devices found -- must research this more !!")
         
         if SMOLAGENTS_AVAILABLE:
             try:
@@ -49,7 +60,8 @@ class BasicAgent:
                 self.model = TransformersModel(
                     model_id="google/gemma-3-4b-it",
                     torch_dtype=torch.float16,
-                    device_map="auto"
+                    device_map="auto",
+                    trust_remote_code=True,
                 )
                 
                 # Create CodeAgent with DuckDuckGoSearchTool (core Unit 2 concept)