geetach commited on
Commit
a216ebf
·
1 Parent(s): 7052543

Consolidate app changes, ensure PNG/LFS removed

Browse files
Files changed (2) hide show
  1. .gitattributes +0 -1
  2. 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={'device': 'mps'}, # Use 'cuda' for GPU, 'cpu' for CPU, 'mps' for M1/M2 Mac
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