Update README.md
Browse files
README.md
CHANGED
@@ -93,11 +93,20 @@ If you use this model in your research, please cite:
|
|
93 |
}
|
94 |
```
|
95 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
96 |
|
97 |
### Usage
|
98 |
|
99 |
<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
|
100 |
|
|
|
101 |
```python
|
102 |
from PIL import Image
|
103 |
import matplotlib.pyplot as plt
|
|
|
93 |
}
|
94 |
```
|
95 |
|
96 |
+
#### Requirements
|
97 |
+
```bash
|
98 |
+
torch
|
99 |
+
torchvision
|
100 |
+
pillow
|
101 |
+
kornia
|
102 |
+
transformers
|
103 |
+
```
|
104 |
|
105 |
### Usage
|
106 |
|
107 |
<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
|
108 |
|
109 |
+
|
110 |
```python
|
111 |
from PIL import Image
|
112 |
import matplotlib.pyplot as plt
|