davanstrien HF Staff commited on
Commit
44beb50
·
1 Parent(s): 3443f84

update cache TTL to 24 hours and increase memory limit to 8GB

Browse files
Files changed (1) hide show
  1. main.py +2 -2
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 = "60"
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="5gb")
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")