Update app.py

app.py

@@ -14,6 +14,7 @@ def segment(image):
     
     with torch.no_grad():
        logits = model(**inputs).logits
+       print(logits)
     
     # model predicts one of the 1000 ImageNet classes
     predicted_label = logits.argmax(-1).item()