Update app.py

app.py

@@ -5,12 +5,23 @@ import matplotlib.pyplot as plt
 import numpy as np
 from PIL import Image
 import tensorflow as tf
-from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
 
-feature_extractor = SegformerFeatureExtractor.from_pretrained(
+# Use a pipeline as a high-level helper
+from transformers import pipeline
+
+pipe = pipeline("image-segmentation", model="mattmdjaga/segformer_b2_clothes")
+
+from transformers import AutoFeatureExtractor, SegformerForSemanticSegmentation
+
+feature_extractor = AutoFeatureExtractor.from_pretrained("mattmdjaga/segformer_b2_clothes")
+model = SegformerForSemanticSegmentation.from_pretrained("mattmdjaga/segformer_b2_clothes")
+
+#from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
+
+#feature_extractor = SegformerFeatureExtractor.from_pretrained(
     "mattmdjaga/segformer_b2_clothes"
 )
-model = TFSegformerForSemanticSegmentation.from_pretrained(
+#model = TFSegformerForSemanticSegmentation.from_pretrained(
     "mattmdjaga/segformer_b2_clothes"
 )