joaodaniel
/

RS-M-CLIP

Zero-Shot Image Classification

Model card Files Files and versions Community

joaodaniel commited on 9 days ago

Commit

902a20c

•

1 Parent(s): 71c9293

Update README.md

Files changed (1) hide show

README.md +2 -0

README.md CHANGED Viewed

@@ -103,6 +103,8 @@ A cute cat                                                     0.0%
 #### Image Classification (in Spanish)
 ```python
 from PIL import Image
 image = preprocess(Image.open('figs/golf_course_004.jpg')).unsqueeze(0)

 #### Image Classification (in Spanish)
 ```python
+model = model.eval()
 from PIL import Image
 image = preprocess(Image.open('figs/golf_course_004.jpg')).unsqueeze(0)