Fix test environment initialisation

app.py

@@ -26,8 +26,10 @@ def main():
     setup_logger()
     load_api_keys()
     
+    is_test = os.environ.get('SPACE_ID') is None
+
     # Initialize the LLM model based on configuration
-    model_id = get_model_id()
+    model_id = get_model_id(is_test=is_test)
     logger.info(f"Initializing with model: {model_id}")
     
     # If using Ollama, we need to specify the API base URL
@@ -48,7 +50,7 @@ def main():
     # Determine if we're in test mode (local) or production (HuggingFace)
     # HuggingFace environment has SPACE_ID environment variable
     
-    is_test = os.environ.get('SPACE_ID') is None
+    
     gradio_config = get_gradio_config(is_test)
     
     # Launch with appropriate configuration