Update README.md
Browse files
README.md
CHANGED
|
@@ -105,6 +105,7 @@ Usage
|
|
| 105 |
|
| 106 |
```python
|
| 107 |
import torch
|
|
|
|
| 108 |
from PIL import Image
|
| 109 |
from model_loader import build_model
|
| 110 |
|
|
@@ -139,6 +140,7 @@ Usage
|
|
| 139 |
|
| 140 |
```python
|
| 141 |
import torch
|
|
|
|
| 142 |
from PIL import Image
|
| 143 |
from model_loader import build_model
|
| 144 |
|
|
|
|
| 105 |
|
| 106 |
```python
|
| 107 |
import torch
|
| 108 |
+
import torchvision
|
| 109 |
from PIL import Image
|
| 110 |
from model_loader import build_model
|
| 111 |
|
|
|
|
| 140 |
|
| 141 |
```python
|
| 142 |
import torch
|
| 143 |
+
import torchvision
|
| 144 |
from PIL import Image
|
| 145 |
from model_loader import build_model
|
| 146 |
|