Spaces:
Build error
Build error
version 1.13, added filter columns, and added a static folder for models, for paster performaance
Browse files- UI_input_output.py +2 -2
UI_input_output.py
CHANGED
|
@@ -189,9 +189,9 @@ def read_clean_model(input_df, mdr_df, model_select):
|
|
| 189 |
print(mdr_df.shape)
|
| 190 |
|
| 191 |
## Model learning ...
|
| 192 |
-
model = SentenceTransformer('all-MiniLM-L6-v2').to(device)
|
| 193 |
# mannually load the model due to not able to connect to huggingface.co at running time from census laptop :
|
| 194 |
-
|
| 195 |
|
| 196 |
if model_select == 'Cosine':
|
| 197 |
result_df = cosineSimilarity(input_df, mdr_df, model)
|
|
|
|
| 189 |
print(mdr_df.shape)
|
| 190 |
|
| 191 |
## Model learning ...
|
| 192 |
+
# model = SentenceTransformer('all-MiniLM-L6-v2').to(device)
|
| 193 |
# mannually load the model due to not able to connect to huggingface.co at running time from census laptop :
|
| 194 |
+
model = SentenceTransformer(f'{model_path}/all-MiniLM-L6-v2').to(device)
|
| 195 |
|
| 196 |
if model_select == 'Cosine':
|
| 197 |
result_df = cosineSimilarity(input_df, mdr_df, model)
|