Update app.py

app.py

@@ -1,8 +1,7 @@
 import gradio as gr
 from txagent import TxAgent
-from tooluniverse import ToolUniverse
-import os
 import logging
+import os
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -13,16 +12,18 @@ class TxAgentApp:
         self.agent = self._initialize_agent()
 
     def _initialize_agent(self):
-        """Initialize the TxAgent with A100 optimizations"""
+        """Initialize the TxAgent with proper parameters"""
         try:
-            logger.info("Initializing TxAgent with A100 optimizations...")
+            logger.info("Initializing TxAgent...")
             agent = TxAgent(
                 model_name="mims-harvard/TxAgent-T1-Llama-3.1-8B",
                 rag_model_name="mims-harvard/ToolRAG-T1-GTE-Qwen2-1.5B",
-                device_map="auto",
-                torch_dtype="auto",
-                enable_xformers=True,
-                max_model_len=8192  # Optimized for A100 80GB
+                # Remove unsupported parameters
+                force_finish=True,
+                enable_checker=True,
+                step_rag_num=10,
+                seed=42,
+                additional_default_tools=["DirectResponse", "RequireClarification"]
             )
             logger.info("Model loading complete")
             return agent
@@ -64,11 +65,7 @@ with gr.Blocks(
         with gr.Column(scale=2):
             chatbot = gr.Chatbot(
                 height=650,
-                bubble_full_width=False,
-                avatar_images=(
-                    "https://example.com/user.png",  # Replace with actual avatars
-                    "https://example.com/bot.png"
-                )
+                bubble_full_width=False
             )
         with gr.Column(scale=1):
             with gr.Accordion("⚙️ Parameters", open=False):
@@ -116,6 +113,5 @@ if __name__ == "__main__":
     ).launch(
         server_name="0.0.0.0",
         server_port=7860,
-        share=False,
-        favicon_path="icon.png"  # Add favicon
+        share=False
     )