Spaces:
Sleeping
Sleeping
Consolidate app changes, ensure PNG/LFS removed
Browse files- .gitattributes +0 -1
- app.py +1 -1
.gitattributes
CHANGED
|
@@ -33,4 +33,3 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
-
*.png filter=lfs diff=lfs merge=lfs -text
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
app.py
CHANGED
|
@@ -78,7 +78,7 @@ if not all_parsed_documents:
|
|
| 78 |
EMBEDDING_MODEL_NAME = "geetach/finetuned-prompt-retriever"
|
| 79 |
embeddings = HuggingFaceEmbeddings(
|
| 80 |
model_name=EMBEDDING_MODEL_NAME,
|
| 81 |
-
model_kwargs={
|
| 82 |
encode_kwargs={'normalize_embeddings': True} # Ensures cosine similarity works correctly
|
| 83 |
)
|
| 84 |
|
|
|
|
| 78 |
EMBEDDING_MODEL_NAME = "geetach/finetuned-prompt-retriever"
|
| 79 |
embeddings = HuggingFaceEmbeddings(
|
| 80 |
model_name=EMBEDDING_MODEL_NAME,
|
| 81 |
+
model_kwargs={}, # Let sentence-transformers auto-detect device (cuda or cpu)
|
| 82 |
encode_kwargs={'normalize_embeddings': True} # Ensures cosine similarity works correctly
|
| 83 |
)
|
| 84 |
|