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 |
|