Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
44beb50
1
Parent(s):
3443f84
update cache TTL to 24 hours and increase memory limit to 8GB
Browse files
main.py
CHANGED
@@ -21,7 +21,7 @@ from datetime import datetime
|
|
21 |
MODEL_NAME = "davanstrien/SmolLM2-360M-tldr-sft-2025-02-12_15-13"
|
22 |
EMBEDDING_MODEL = "nomic-ai/modernbert-embed-base"
|
23 |
BATCH_SIZE = 2000
|
24 |
-
CACHE_TTL = "
|
25 |
TRENDING_CACHE_TTL = "900" # 15 minutes cache for trending data
|
26 |
|
27 |
if torch.cuda.is_available():
|
@@ -48,7 +48,7 @@ if sys.platform == "darwin":
|
|
48 |
LOCAL = True
|
49 |
DATA_DIR = "data" if LOCAL else "/data"
|
50 |
# Configure cache
|
51 |
-
cache.setup("mem://", size_limit="
|
52 |
|
53 |
# Initialize ChromaDB client
|
54 |
client = chromadb.PersistentClient(path=f"{DATA_DIR}/chroma")
|
|
|
21 |
MODEL_NAME = "davanstrien/SmolLM2-360M-tldr-sft-2025-02-12_15-13"
|
22 |
EMBEDDING_MODEL = "nomic-ai/modernbert-embed-base"
|
23 |
BATCH_SIZE = 2000
|
24 |
+
CACHE_TTL = "24h"
|
25 |
TRENDING_CACHE_TTL = "900" # 15 minutes cache for trending data
|
26 |
|
27 |
if torch.cuda.is_available():
|
|
|
48 |
LOCAL = True
|
49 |
DATA_DIR = "data" if LOCAL else "/data"
|
50 |
# Configure cache
|
51 |
+
cache.setup("mem://", size_limit="8gb")
|
52 |
|
53 |
# Initialize ChromaDB client
|
54 |
client = chromadb.PersistentClient(path=f"{DATA_DIR}/chroma")
|