Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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()
|