Spaces:
Sleeping
Sleeping
Update webchat.py
Browse files- webchat.py +2 -2
webchat.py
CHANGED
|
@@ -88,8 +88,8 @@ cache_dir = os.path.join(current_dir, ".cache")
|
|
| 88 |
if not os.path.exists(cache_dir):
|
| 89 |
os.makedirs(cache_dir)
|
| 90 |
# Download the model (specify the correct model identifier)
|
| 91 |
-
|
| 92 |
-
model_name = "all-MiniLM-L6-v2"
|
| 93 |
model = SentenceTransformer(model_name, cache_folder=cache_dir)
|
| 94 |
# Print confirmation message
|
| 95 |
print(f"Model '{model_name}' downloaded and loaded from cache directory: {cache_dir}")
|
|
|
|
| 88 |
if not os.path.exists(cache_dir):
|
| 89 |
os.makedirs(cache_dir)
|
| 90 |
# Download the model (specify the correct model identifier)
|
| 91 |
+
model_name = 'sentence-transformers/all-MiniLM-L6-v2'
|
| 92 |
+
#model_name = "all-MiniLM-L6-v2"
|
| 93 |
model = SentenceTransformer(model_name, cache_folder=cache_dir)
|
| 94 |
# Print confirmation message
|
| 95 |
print(f"Model '{model_name}' downloaded and loaded from cache directory: {cache_dir}")
|