swaroop77 commited on
Commit
b2cd6c1
·
verified ·
1 Parent(s): a18f083

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ logging.basicConfig(level=logging.INFO)
13
  # --- Initialize Models (Load these once) ---
14
  # Ensure model path is accessible, default path works in Docker
15
  try:
16
- embedding_model = SentenceTransformer('all-MiniLM-L6-v2')
17
  logging.info("SentenceTransformer model loaded successfully.")
18
  except Exception as e:
19
  logging.error(f"Error loading SentenceTransformer model: {e}")
 
13
  # --- Initialize Models (Load these once) ---
14
  # Ensure model path is accessible, default path works in Docker
15
  try:
16
+ embedding_model = SentenceTransformer('all-MiniLM-L6-v2', from_tf=True)
17
  logging.info("SentenceTransformer model loaded successfully.")
18
  except Exception as e:
19
  logging.error(f"Error loading SentenceTransformer model: {e}")