Update app.py

app.py

@@ -4,19 +4,16 @@ import logging
 import torch
 from txagent import TxAgent
 import gradio as gr
-from huggingface_hub import hf_hub_download
 from tooluniverse import ToolUniverse
-from tqdm import tqdm
 
-# Configuration - Now using remote Hugging Face models
+# Configuration - Using remote Hugging Face models
 CONFIG = {
     "model_name": "mims-harvard/TxAgent-T1-Llama-3.1-8B",
-    "rag_model_name": "mims-harvard/ToolRAG-T1-GTE-Qwen2-1.5B", 
+    "rag_model_name": "mims-harvard/ToolRAG-T1-GTE-Qwen2-1.5B",
     "embedding_filename": "ToolRAG-T1-GTE-Qwen2-1.5Btool_embedding_47dc56b3e3ddeb31af4f19defdd538d984de1500368852a0fab80bc2e826c944.pt",
     "tool_files": {
         "new_tool": "./data/new_tool.json"
-    },
-    "load_from_hub": True  # Flag to load directly from Hugging Face
+    }
 }
 
 # Logging setup
@@ -47,7 +44,7 @@ def load_embeddings(agent):
         except Exception as e:
             logger.error(f"Failed to load embeddings: {e}")
     
-    logger.info("Generating tool embeddings from remote model...")
+    logger.info("Generating tool embeddings...")
     try:
         tools = agent.tooluniverse.get_all_tools()
         descriptions = [tool["description"] for tool in tools]
@@ -71,7 +68,7 @@ class TxAgentApp:
         try:
             logger.info("Initializing TxAgent with remote models...")
             
-            # Initialize with remote models
+            # Initialize without local_files_only parameter
             self.agent = TxAgent(
                 CONFIG["model_name"],
                 CONFIG["rag_model_name"],
@@ -80,11 +77,10 @@ class TxAgentApp:
                 enable_checker=True,
                 step_rag_num=10,
                 seed=100,
-                additional_default_tools=["DirectResponse", "RequireClarification"],
-                local_files_only=False  # Force loading from Hugging Face Hub
+                additional_default_tools=["DirectResponse", "RequireClarification"]
             )
             
-            logger.info("Loading remote models...")
+            logger.info("Loading models from Hugging Face Hub...")
             self.agent.init_model()
             
             logger.info("Preparing embeddings...")