Preetham Ganesh
commited on
Commit
·
b82118e
1
Parent(s):
4e08061
updated README. updated usage code.
Browse files
README.md
CHANGED
|
@@ -60,7 +60,7 @@ input_data = tf.ones((1, 28, 28, 1), dtype=tf.float32)
|
|
| 60 |
output = model(input_data)["output_0"]
|
| 61 |
|
| 62 |
# Prints the predicted class (e.g., 0 for normal, 1 for abnormal)
|
| 63 |
-
predicted_digit = tf.argmax(output, axis=-1).numpy()
|
| 64 |
print("Predicted digit: ", predicted_digit)
|
| 65 |
```
|
| 66 |
|
|
|
|
| 60 |
output = model(input_data)["output_0"]
|
| 61 |
|
| 62 |
# Prints the predicted class (e.g., 0 for normal, 1 for abnormal)
|
| 63 |
+
predicted_digit = tf.argmax(output, axis=-1).numpy()[0]
|
| 64 |
print("Predicted digit: ", predicted_digit)
|
| 65 |
```
|
| 66 |
|