Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -13,8 +13,8 @@ def predict(img):
|
|
| 13 |
#img_array = tf.expand_dims(img_array, 0)
|
| 14 |
|
| 15 |
predictions = model.predict(img_array)
|
| 16 |
-
predicted_class = class_names[np.argmax(predictions[0])]
|
| 17 |
-
confidence = round(100 * (np.max(predictions[0])), 2)
|
| 18 |
return {class_names[i]: float(predictions[0][i]) for i in range(3)}
|
| 19 |
|
| 20 |
|
|
|
|
| 13 |
#img_array = tf.expand_dims(img_array, 0)
|
| 14 |
|
| 15 |
predictions = model.predict(img_array)
|
| 16 |
+
# predicted_class = class_names[np.argmax(predictions[0])]
|
| 17 |
+
# confidence = round(100 * (np.max(predictions[0])), 2)
|
| 18 |
return {class_names[i]: float(predictions[0][i]) for i in range(3)}
|
| 19 |
|
| 20 |
|