Spaces:
Build error
Build error
Commit ·
9777e8f
1
Parent(s): c29f649
corrected # of classes
Browse files
app.py
CHANGED
|
@@ -44,7 +44,7 @@ def classify_image(inp):
|
|
| 44 |
img_array = tf.expand_dims(img_array, 0) # Creates a batch axis
|
| 45 |
|
| 46 |
predictions = model.predict(img_array, verbose=0).flatten()
|
| 47 |
-
confidences = {breed_names_norm[i]: float(predictions[i]) for i in range(
|
| 48 |
|
| 49 |
return confidences
|
| 50 |
|
|
|
|
| 44 |
img_array = tf.expand_dims(img_array, 0) # Creates a batch axis
|
| 45 |
|
| 46 |
predictions = model.predict(img_array, verbose=0).flatten()
|
| 47 |
+
confidences = {breed_names_norm[i]: float(predictions[i]) for i in range(120)}
|
| 48 |
|
| 49 |
return confidences
|
| 50 |
|