test

app.py

@@ -7,13 +7,17 @@ from PIL import Image
 import tensorflow as tf
 from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
 
+# Load feature extractor
 feature_extractor = SegformerFeatureExtractor.from_pretrained(
     "jakka/segformer-b0-finetuned-warehouse-part-1-V2"
 )
+
+# Load model using PyTorch weights
 model = TFSegformerForSemanticSegmentation.from_pretrained(
-    "jakka/segformer-b0-finetuned-warehouse-part-1-V2"
+    "jakka/segformer-b0-finetuned-warehouse-part-1-V2", from_pt=True
 )
 
+
 def ade_palette():
     """ADE20K palette that maps each class to RGB values."""
     return [