Spaces:
Sleeping
Sleeping
root
commited on
Commit
·
6f68660
1
Parent(s):
f709fa8
Erase display function
Browse files
app.py
CHANGED
|
@@ -10,9 +10,6 @@ def classifyArchitecture(img):
|
|
| 10 |
inputs = image_processor(img, return_tensors="pt")
|
| 11 |
output = classifier(**inputs)
|
| 12 |
|
| 13 |
-
# Display the image
|
| 14 |
-
display(img)
|
| 15 |
-
|
| 16 |
# Predicted Class probabilities
|
| 17 |
proba = output.logits.softmax(1)
|
| 18 |
preds = proba.argmax(1)
|
|
|
|
| 10 |
inputs = image_processor(img, return_tensors="pt")
|
| 11 |
output = classifier(**inputs)
|
| 12 |
|
|
|
|
|
|
|
|
|
|
| 13 |
# Predicted Class probabilities
|
| 14 |
proba = output.logits.softmax(1)
|
| 15 |
preds = proba.argmax(1)
|