Spaces:
Sleeping
Sleeping
Commit
·
7fdcf83
1
Parent(s):
bde552e
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,6 +8,7 @@ def classify_image(inp):
|
|
| 8 |
inp = inp.reshape((-1, 224, 224, 3))
|
| 9 |
inp == inp/255
|
| 10 |
prediction = model.predict(inp)[0]
|
|
|
|
| 11 |
confidences = {model.config['id2label'][str(i)]: float(prediction[i]) for i in range(101)}
|
| 12 |
return confidences
|
| 13 |
|
|
|
|
| 8 |
inp = inp.reshape((-1, 224, 224, 3))
|
| 9 |
inp == inp/255
|
| 10 |
prediction = model.predict(inp)[0]
|
| 11 |
+
print(prediction, model.config)
|
| 12 |
confidences = {model.config['id2label'][str(i)]: float(prediction[i]) for i in range(101)}
|
| 13 |
return confidences
|
| 14 |
|