Spaces:
Running
Running
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}")
|