Skip to content

HuggingFace app set local model or personal model on HuggingFace

How can one set a personal trained model in the demo app?

In config.yaml if I set my model as model_name: doc_ufcn_samaritan_v1 or model_name: johnlockejrr/doc_ufcn_samaritan_v1 it tries to get https://huggingface.co/Teklia/doc-ufcn-doc_ufcn_samaritan_v1.

How can I set my personal repo?

Temporarily solved editing the file doc_ufcn/models.py

HUGGING_FACE_REPO_PREFIX = "johnlockejrr/"

Should be this a config option?

Edited by Teodor Bors