Update README.md

README.md

@@ -1,5 +1,6 @@
 ---
 license: apache-2.0
+pipeline_tag: image-classification
 ---
 # Model Card: Fine-Tuned Vision Transformer (ViT) for NSFW Image Classification
 
@@ -32,12 +33,15 @@ classifier(image)
 ``` markdown
 
 # Load model directly
+import torch
+from PIL import Image
 from transformers import AutoModelForImageClassification, ViTImageProcessor
 
+img = Image.open("<path_to_image_file>")
 model = AutoModelForImageClassification.from_pretrained("RealFalconsAI/nsfw_image_detection")
 processor = ViTImageProcessor.from_pretrained('RealFalconsAI/nsfw_image_detection')
 with torch.no_grad():
-    inputs = processor(images=<image>, return_tensors="pt")
+    inputs = processor(images=img, return_tensors="pt")
     outputs = model(**inputs)
     logits = outputs.logits