Update app.py

app.py

@@ -5,12 +5,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-ade-640-640"
 )
-model = TFSegformerForSemanticSegmentation.from_pretrained(
+model = SegformerForSemanticSegmentation.from_pretrained(
     "nvidia/segformer-b5-finetuned-ade-640-640"
 )