Update README.md
Browse files
README.md
CHANGED
|
@@ -103,6 +103,8 @@ A cute cat 0.0%
|
|
| 103 |
|
| 104 |
#### Image Classification (in Spanish)
|
| 105 |
```python
|
|
|
|
|
|
|
| 106 |
from PIL import Image
|
| 107 |
image = preprocess(Image.open('figs/golf_course_004.jpg')).unsqueeze(0)
|
| 108 |
|
|
|
|
| 103 |
|
| 104 |
#### Image Classification (in Spanish)
|
| 105 |
```python
|
| 106 |
+
model = model.eval()
|
| 107 |
+
|
| 108 |
from PIL import Image
|
| 109 |
image = preprocess(Image.open('figs/golf_course_004.jpg')).unsqueeze(0)
|
| 110 |
|