Spaces:

zergswim
/

test

Runtime error

zergswim commited on Sep 30, 2022

Commit

f03a370

1 Parent(s): 51012d6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

     with torch.no_grad():
        logits = model(**inputs).logits
+       print(logits)
     # model predicts one of the 1000 ImageNet classes
     predicted_label = logits.argmax(-1).item()