anon5 commited on
Commit
e5de65c
1 Parent(s): 29ccfe0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -28,8 +28,6 @@ def hello_world():
28
  inputs = feature_extractor(images=image, return_tensors="pt")
29
  outputs = model(**inputs)
30
  logits = outputs.logits
31
-
32
- print(logits)
33
 
34
  # model predicts one of the 1000 ImageNet classes
35
  predicted_class_idx = logits.argmax(-1).item()
 
28
  inputs = feature_extractor(images=image, return_tensors="pt")
29
  outputs = model(**inputs)
30
  logits = outputs.logits
 
 
31
 
32
  # model predicts one of the 1000 ImageNet classes
33
  predicted_class_idx = logits.argmax(-1).item()