Update README.md

README.md

@@ -5,7 +5,7 @@ library_name: transformers
 pipeline_tag: image-classification
 ---
 
-# SwiftFormer
+# SwiftFormer (swiftformer-s)
 
 ## Model description
 
@@ -28,12 +28,12 @@ SwiftFormer paper introduces a novel efficient additive attention mechanism that
     image = Image.open(requests.get(url, stream=True).raw)
     
     from transformers import ViTImageProcessor
-    processor = ViTImageProcessor.from_pretrained('shehan97/swiftformer-xs')
+    processor = ViTImageProcessor.from_pretrained('shehan97/swiftformer-s')
     inputs = processor(images=image, return_tensors="pt")
 
 
     from transformers.models.swiftformer import SwiftFormerForImageClassification
-    new_model = SwiftFormerForImageClassification.from_pretrained('shehan97/swiftformer-xs')
+    new_model = SwiftFormerForImageClassification.from_pretrained('shehan97/swiftformer-s')
     
     output = new_model(inputs['pixel_values'], output_hidden_states=True)
     logits = output.logits