Ali2206 commited on
Commit
7f5790c
·
verified ·
1 Parent(s): 02ebb35

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -44,6 +44,9 @@ os.environ.update({
44
  "TOKENIZERS_PARALLELISM": "false",
45
  "CUDA_LAUNCH_BLOCKING": "1"
46
  })
 
 
 
47
  from txagent.txagent import TxAgent
48
  # ==================== UTILITY FUNCTIONS ====================
49
  def sanitize_text(text: str) -> str:
 
44
  "TOKENIZERS_PARALLELISM": "false",
45
  "CUDA_LAUNCH_BLOCKING": "1"
46
  })
47
+ current_dir = os.path.dirname(os.path.abspath(__file__))
48
+ src_path = os.path.abspath(os.path.join(current_dir, "src"))
49
+ sys.path.insert(0, src_path)
50
  from txagent.txagent import TxAgent
51
  # ==================== UTILITY FUNCTIONS ====================
52
  def sanitize_text(text: str) -> str: