Skip to content

Commit fb50f06

Browse files
Update README.md
1 parent 54abed9 commit fb50f06

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

README.md

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,18 @@ conda install pytorch==2.0.1 torchvision==0.15.2 pytorch-cuda=11.7 -c pytorch -c
1515
pip install numpy opencv-python
1616
```
1717

18+
## Inference with HuggingFace 🤗
19+
Install huggingface-hub
20+
```
21+
pip install huggingface-hub
22+
```
23+
Inference uses our pretrained model, trained with multiple datasets, neuflow_mixed.pth:
24+
```
25+
python infer_hf.py
26+
```
27+
28+
<img src="example_result.jpg" width="400" >
29+
1830
## Inference
1931

2032
Inference uses our pretrained model, trained with multiple datasets, neuflow_mixed.pth:

0 commit comments

Comments
 (0)