Update app.py
Browse files
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}")
|