Update app.py

app.py

@@ -4,7 +4,7 @@ import logging
 import torch
 from txagent import TxAgent
 import gradio as gr
-from huggingface_hub import snapshot_download
+from huggingface_hub import hf_hub_download, snapshot_download
 from tooluniverse import ToolUniverse
 
 # Configuration
@@ -50,10 +50,8 @@ def download_model_files():
 
 def generate_embeddings(agent):
     embedding_path = CONFIG["embedding_filename"]
-
     if os.path.exists(embedding_path):
-        print("Embeddings file already exists — loading...")
-        agent.rag_model.tool_desc_embedding = torch.load(embedding_path)
+        print("✅ Using pre-uploaded embeddings file")
         return
 
     print("Generating missing tool embeddings...")
@@ -149,4 +147,4 @@ if __name__ == "__main__":
     prepare_tool_files()
     download_model_files()
     interface = create_interface()
-    interface.launch(server_name="0.0.0.0", server_port=7860, share=False)
+    interface.launch(server_name="0.0.0.0", server_port=7860, share=False)