sujoy0011 commited on
Commit
219c095
·
verified ·
1 Parent(s): 82d4efa

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -67,7 +67,8 @@ class NewsAgent:
67
  """Initialize the embedding model."""
68
  try:
69
  self.embedding_model = HuggingFaceEmbeddings(
70
- model_name="sentence-transformers/all-MiniLM-L6-v2"
 
71
  )
72
  print("✅ Embedding model initialized")
73
  except Exception as e:
 
67
  """Initialize the embedding model."""
68
  try:
69
  self.embedding_model = HuggingFaceEmbeddings(
70
+ model_name="sentence-transformers/all-MiniLM-L6-v2",
71
+ cache_folder="/app/cache"
72
  )
73
  print("✅ Embedding model initialized")
74
  except Exception as e: