FredZhang7
/

google-safesearch-mini-tfjs

Image Classification

Model card Files Files and versions

FredZhang7 commited on Dec 27, 2022

Commit

312519c

·

1 Parent(s): 6fcab6a

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="c9c0cc6b7e93796a712317baa7ca70db9e939688")
 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="6fcab6a27595a5f008625ec88c77c10739f3c219")
 PATH_TO_IMAGE = 'https://images.unsplash.com/photo-1594568284297-7c64464062b1'
 PRINT_TENSOR = False