openrecall / openrecall

OpenRecall is a fully open-source, privacy-first alternative to proprietary solutions like Microsoft's Windows Recall. With OpenRecall, you can easily access your digital history, enhancing your memory and productivity without compromising your privacy.
GNU Affero General Public License v3.0
1.47k stars 65 forks source link

[Feature Request] Allow users to select different models for OCR and NLP #46

Open hiteshseth opened 2 weeks ago

hiteshseth commented 2 weeks ago

Currently code is using following models which are hard coded. Can we give users an option to select different models if they prefer by specifying a location where model would be stored. This way people can upgrade their models or use domain specific models depending on their needs

NLP-
model = SentenceTransformer("all-MiniLM-L6-v2")

OCR- det_arch="db_mobilenet_v3_large", reco_arch="crnn_mobilenet_v3_large",