aapot
commited on
Commit
•
86de605
1
Parent(s):
4d5d953
Update README
Browse files- README.md +1 -0
- architecture.png +0 -0
README.md
CHANGED
|
@@ -38,6 +38,7 @@ This model is intended to be used for analyzing whether a pair of YouTube videos
|
|
| 38 |
As this model is a custom PyTorch model, not normal transformers model, you need to clone this model repository first. The repository contains model code in `RRUM` class (RRUM stands for RegretsReporter Unified Model) in `unifiedmodel.py` file. For loading the model from Hugging Face model hub, there also is a Hugging Face model wrapper named `YoutubeVideoSimilarityModel` in `huggingface_model_wrapper.py` file. Needed Python requirements are specified in `requirements.txt` file. To load the model, follow these steps:
|
| 39 |
1. `git clone https://huggingface.co/mozilla-foundation/youtube_video_similarity_model_wt`
|
| 40 |
2. `pip install -r requirements.txt`
|
|
|
|
| 41 |
And finally load the model with the following example code:
|
| 42 |
```python
|
| 43 |
from huggingface_model_wrapper import YoutubeVideoSimilarityModel
|
|
|
|
| 38 |
As this model is a custom PyTorch model, not normal transformers model, you need to clone this model repository first. The repository contains model code in `RRUM` class (RRUM stands for RegretsReporter Unified Model) in `unifiedmodel.py` file. For loading the model from Hugging Face model hub, there also is a Hugging Face model wrapper named `YoutubeVideoSimilarityModel` in `huggingface_model_wrapper.py` file. Needed Python requirements are specified in `requirements.txt` file. To load the model, follow these steps:
|
| 39 |
1. `git clone https://huggingface.co/mozilla-foundation/youtube_video_similarity_model_wt`
|
| 40 |
2. `pip install -r requirements.txt`
|
| 41 |
+
|
| 42 |
And finally load the model with the following example code:
|
| 43 |
```python
|
| 44 |
from huggingface_model_wrapper import YoutubeVideoSimilarityModel
|
architecture.png
CHANGED
|
|