Update app.py

app.py

@@ -15,7 +15,7 @@ def segment(image):
     with torch.no_grad():
        logits = model(**inputs).logits
        probs = torch.nn.Softmax(dim=1)(logits)
-       labels = [(prob, model.config.id2label[idx]) for idx, prob in enumerate(probs)]
+       labels = [prob, model.config.id2label[idx] for idx, prob in enumerate(probs)]
        print(labels)
     
     # model predicts one of the 1000 ImageNet classes