FredZhang7
/

google-safesearch-mini-tfjs

Image Classification

Model card Files Files and versions

FredZhang7 commited on Dec 27, 2022

Commit

a0de6a5

·

1 Parent(s): d16bbb5

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -22,7 +22,7 @@ pip install --upgrade transformers torchvision
 from transformers import AutoModelForImageClassification
 from torch import cuda
-model = AutoModelForImageClassification.from_pretrained("FredZhang7/google-safesearch-mini", trust_remote_code=True, revision="d0b4c6be6d908c39c0dd83d25dce50c0e861e46a")
 PATH_TO_IMAGE = 'https://images.unsplash.com/photo-1594568284297-7c64464062b1'
 PRINT_TENSOR = False

 from transformers import AutoModelForImageClassification
 from torch import cuda
+model = AutoModelForImageClassification.from_pretrained("FredZhang7/google-safesearch-mini", trust_remote_code=True, revision="d16bbb53644959a071bd3b8ca0d886c65de6bd45")
 PATH_TO_IMAGE = 'https://images.unsplash.com/photo-1594568284297-7c64464062b1'
 PRINT_TENSOR = False