Update app.py
Browse files
app.py
CHANGED
@@ -17,6 +17,8 @@ def image_classifier(inp):
|
|
17 |
score = tf.nn.softmax(predictions)
|
18 |
r = {}
|
19 |
print(score)
|
|
|
|
|
20 |
for class_name in class_names:
|
21 |
r[class_name] = score[0][class_names.index(class_name)]
|
22 |
return r
|
|
|
17 |
score = tf.nn.softmax(predictions)
|
18 |
r = {}
|
19 |
print(score)
|
20 |
+
for s in score:
|
21 |
+
print(s)
|
22 |
for class_name in class_names:
|
23 |
r[class_name] = score[0][class_names.index(class_name)]
|
24 |
return r
|