TorchScript converted model with synset

convert.py

@@ -0,0 +1,18 @@
+from transformers import AutoImageProcessor, AutoModelForImageClassification
+import torch
+from PIL import Image
+from transformers import AutoTokenizer
+
+model_name = "DenisNovac/nsfw_image_detection"
+
+model = AutoModelForImageClassification.from_pretrained(model_name, torchscript=True, return_dict=False)
+
+processor = AutoImageProcessor.from_pretrained(model_name)
+
+image = Image.open("images/hentai.jpg")
+image_inputs = processor(images=image, return_tensors="pt")
+
+config = {'forward': [image_inputs['pixel_values']]}
+converted = torch.jit.trace_module(model,  config)
+
+torch.jit.save(converted, "converted-to-torchscript.pt")

converted-to-torchscript.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4f80aebd382cedcb4213eda71479edd851d54ae7f8d5ddf96adc213689dcc412
+size 343559956

synset.txt

@@ -0,0 +1,2 @@
+normal
+nsfw