Where is the binary intent classifier model?
TCL606 opened this issue · comments
I tried to run the r3 demo. I found that I needed a binary intent classifier model, but I did not find one. Could you please tell me where the model is? It would be great if you could give a documentation to run the r3demo. Thank you!
- To run the R3 demo, we need three models: revision model, binary intent classifier, and intent classifier.
# code/R3_In2Writing2022/baseline_model.py#L107-L110
class BaselineModel:
def __init__(self, revision_model_path: str = '../models/2021-01-06-multi-intent',
binary_classifier_model_path: str = '../models/binary_intent_classifier',
intent_classifier_model_path: str = '../models/intent_classifier_src'):
- These models can be downloaded from Hugging Face Hub:
- Make sure that the path points to the downloaded directory of the models, and that they are loaded correctly.
- Then install
pip install -r requirements.txt
and runstreamlit run main.py
- More details on streamlit, here.