Update app.py
Browse files
app.py
CHANGED
|
@@ -50,7 +50,11 @@ os.environ.update({
|
|
| 50 |
"TOKENIZERS_PARALLELISM": "false",
|
| 51 |
"CUDA_LAUNCH_BLOCKING": "1"
|
| 52 |
})
|
|
|
|
|
|
|
|
|
|
| 53 |
|
|
|
|
| 54 |
# Initialize cache with 10GB limit
|
| 55 |
cache = Cache(file_cache_dir, size_limit=10 * 1024**3)
|
| 56 |
|
|
|
|
| 50 |
"TOKENIZERS_PARALLELISM": "false",
|
| 51 |
"CUDA_LAUNCH_BLOCKING": "1"
|
| 52 |
})
|
| 53 |
+
current_dir = os.path.dirname(os.path.abspath(__file__))
|
| 54 |
+
src_path = os.path.abspath(os.path.join(current_dir, "src"))
|
| 55 |
+
sys.path.insert(0, src_path)
|
| 56 |
|
| 57 |
+
from txagent.txagent import TxAgent
|
| 58 |
# Initialize cache with 10GB limit
|
| 59 |
cache = Cache(file_cache_dir, size_limit=10 * 1024**3)
|
| 60 |
|