original tools

app.py

@@ -14,18 +14,10 @@ from medrax.agent import *
 from medrax.tools import *
 from medrax.utils import *
 
-from huggingface_hub import login
-import os
-
 warnings.filterwarnings("ignore")
 logging.set_verbosity_error()
 _ = load_dotenv()
 
-
-hf_token = os.getenv("HUGGINGFACE_TOKEN")
-if hf_token:
-    login(token=hf_token)
-
 def initialize_agent(
     prompt_file,
     tools_to_use=None,
@@ -38,7 +30,6 @@ def initialize_agent(
     openai_kwargs={}
 ):
     """Initialize the MedRAX agent with specified tools and configuration.
-
     Args:
         prompt_file (str): Path to file containing system prompts
         tools_to_use (List[str], optional): List of tool names to initialize. If None, all tools are initialized.
@@ -49,7 +40,6 @@ def initialize_agent(
         temperature (float, optional): Temperature for the model. Defaults to 0.7.
         top_p (float, optional): Top P for the model. Defaults to 0.95.
         openai_kwargs (dict, optional): Additional keyword arguments for OpenAI API, such as API key and base URL.
-
     Returns:
         Tuple[Agent, Dict[str, BaseTool]]: Initialized agent and dictionary of tool instances
     """
@@ -111,9 +101,9 @@ if __name__ == "__main__":
         "ChestXRaySegmentationTool",
         "ChestXRayReportGeneratorTool",
         "XRayVQATool",
+        # "LlavaMedTool",
         # "XRayPhraseGroundingTool",
         # "ChestXRayGeneratorTool",
-        # "LlavaMedTool",
     ]
 
     # Collect the ENV variables