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()
|