1

README.md

@@ -9,4 +9,23 @@ app_file: app.py
 pinned: false
 ---
 
+```python
+from transformers import SegformerFeatureExtractor, SegformerForSemanticSegmentation
+from PIL import Image
+import requests
+
+feature_extractor = SegformerFeatureExtractor.from_pretrained("nvidia/segformer-b3-finetuned-cityscapes-1024-1024")
+model = SegformerForSemanticSegmentation.from_pretrained("nvidia/segformer-b3-finetuned-cityscapes-1024-1024")
+
+url = ["https://imagescdn.gettyimagesbank.com/500/201407/a8281860.jpg",
+       "https://imagescdn.gettyimagesbank.com/500/14/890/614/3/a8281872.jpg",
+       "https://50plus.or.kr/upload/im/2019/06/d3007817-9c63-468d-bbc1-0afa8f10874e.jpg"
+       ]
+image = Image.open(requests.get(url, stream=True).raw)
+
+inputs = feature_extractor(images=image, return_tensors="pt")
+outputs = model(**inputs)
+logits = outputs.logits  # shape (batch_size, num_labels, height/4, width/4)
+```
+
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

config.json

@@ -7,8 +7,8 @@
   "decoder_hidden_size": 768,
   "depths": [
     3,
-    6,
-    40,
+    4,
+    18,
     3
   ],
   "downsampling_rates": [

preprocessor_config.json

@@ -14,5 +14,5 @@
   ],
   "reduce_labels": false,
   "resample": 2,
-  "size": 1024
+  "size": 512
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:bdefc7b5bf4b548d347e6bcbe3ab1a0ff4370047211ae9e56c9e53dfb1f78033
-size 338896453
+oid sha256:4b3916b8781255eaa6b596be75eb4610001e2ff18083a4aa897539e1d1f41f3f
+size 189204985

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e210af2d6d0dfb7eabecfcb1831845830b762a04d65d69e879cfc17288a11540
-size 339967916
+oid sha256:c522e7c5c3416e9c2009fbd90cd6e36912b8d9d8b1cd0c591e1be4daadbfb343
+size 189787044