Upload 4 files

README.md

@@ -1,5 +1,5 @@
 ---
-title: segformer-b5-finetuned-cityscapes-1024-1024
+title: mymy
 emoji: 💻
 colorFrom: blue
 colorTo: red

app.py

@@ -18,12 +18,12 @@ import matplotlib.pyplot as plt
 import numpy as np
 from PIL import Image
 import tensorflow as tf
-from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
+from transformers import SegformerFeatureExtractor, SegformerForSemanticSegmentation
 
 feature_extractor = SegformerFeatureExtractor.from_pretrained(
     "nvidia/segformer-b5-finetuned-cityscapes-1024-1024"
 )
-model = TFSegformerForSemanticSegmentation.from_pretrained(
+model = SegformerForSemanticSegmentation.from_pretrained(
     "nvidia/segformer-b5-finetuned-cityscapes-1024-1024"
 )