remove map auto

app.py

@@ -129,7 +129,7 @@ class GaiaAgent:
 
         # Initialize the LLM
         logging.info("Loading LLM... This may take a few minutes on first startup.")
-        # Using a smaller, CPU-friendly model to avoid memory issues on Hugging Face Spaces
+        # Removing device_map for better compatibility with ZeroGPU environments
         llm = HuggingFacePipeline.from_model_id(
             model_id="microsoft/Phi-3-mini-4k-instruct",
             task="text-generation",
@@ -141,7 +141,6 @@ class GaiaAgent:
             },
             torch_dtype="auto",
             trust_remote_code=True,  # Required for Phi-3
-            device_map="auto",
         )
         logging.info("LLM loaded successfully.")