Update app.py

app.py

@@ -6,15 +6,13 @@ import numpy as np
 
 extractor = AutoFeatureExtractor.from_pretrained("hshetty/segmentation-model-finetuned-on-semantic-sidewalk-3e-4-e5")
 model = SegformerForSemanticSegmentation.from_pretrained("hshetty/segmentation-model-finetuned-on-semantic-sidewalk-3e-4-e5")
-counter = 0
+
 def classify(im):
   inputs = extractor(images=im, return_tensors="pt")
   outputs = model(**inputs)
   logits = outputs.logits
   classes = logits[0].detach().cpu().numpy().argmax(axis=0)
   colors = np.array([[128,0,0], [128,128,0], [0, 0, 128], 	[128,0,128], [0, 0, 0]])
-  counter+=1
-  print("hshetty:{}".format(counter))
   return colors[classes]