remove upscale to original res - doesn't work on :hugging: spaces yet

app.py

@@ -25,7 +25,8 @@ feature_extractor = SegformerFeatureExtractor.from_pretrained(model_name)
 model = SegformerForSemanticSegmentation.from_pretrained(model_name)
 
 def inference(image):
-    inputs = feature_extractor(images=image.resize((1024,1024)), return_tensors="pt")
+    image = image.resize((1024,1024))
+    inputs = feature_extractor(images=image, return_tensors="pt")
     outputs = model(**inputs)
 
     # First, rescale logits to original image size