Hugging Face's logo

Spaces:
xferdie
/
ViT
Paused

App Files Community
xferdie commited on
Commit
eeab7a4
·
verified ·
1 Parent(s): 8a99a61

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -14,4 +14,3 @@ logits = outputs.logits
14
  # model predicts one of the 1000 ImageNet classes
15
  predicted_class_idx = logits.argmax(-1).item()
16
  print("Predicted class:", model.config.id2label[predicted_class_idx])
17
-
 
14
  # model predicts one of the 1000 ImageNet classes
15
  predicted_class_idx = logits.argmax(-1).item()
16
  print("Predicted class:", model.config.id2label[predicted_class_idx])