Update README.md

README.md

@@ -6,6 +6,32 @@ pipeline_tag: image-classification
 Copyright: Falcons.ai
 Forked from: https://huggingface.co/Falconsai/nsfw_image_detection/tree/main
 
+DJL project that uses the model: https://github.com/Baklanov-Soft/image-hosting-processing
+
+`converted-to-torchscript.pt` - converted to TorchScript (to be used in DJL) by script:
+
+```python
+from PIL import Image
+from transformers import AutoTokenizer
+
+model_name = "Falconsai/nsfw_image_detection"
+
+model = AutoModelForImageClassification.from_pretrained(model_name, torchscript=True, return_dict=False)
+
+processor = AutoImageProcessor.from_pretrained(model_name)
+
+# example nsfw picture to trigger the model, find it yourself
+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.pt")
+```
+
+
+
 # Model Card: Fine-Tuned Vision Transformer (ViT) for NSFW Image Classification
 
 ## Model Description