Update app.py

app.py

@@ -46,7 +46,11 @@ class BasicAgent:
         if SMOLAGENTS_AVAILABLE:
             try:
                 # Initialize the model (from Unit 2)
-                self.model = TransformersModel(model_id="google/gemma-3-4b-it", torch_dtype=torch.float16)
+                self.model = TransformersModel(
+                    model_id="google/gemma-3-4b-it",
+                    torch_dtype=torch.float16,
+                    device_map="auto"
+                )
                 
                 # Create CodeAgent with DuckDuckGoSearchTool (core Unit 2 concept)
                 self.agent = CodeAgent(