Update README.md
Browse files
README.md
CHANGED
|
@@ -44,3 +44,6 @@ pipeline_tag: image-classification
|
|
| 44 |
predictions = np.argmax(logits, axis=1)
|
| 45 |
predicted_labels = [labels[prediction] for prediction in predictions]
|
| 46 |
print(predicted_labels)
|
|
|
|
|
|
|
|
|
|
|
|
| 44 |
predictions = np.argmax(logits, axis=1)
|
| 45 |
predicted_labels = [labels[prediction] for prediction in predictions]
|
| 46 |
print(predicted_labels)
|
| 47 |
+
|
| 48 |
+
## Model generation
|
| 49 |
+
The model has been created using the 'train_shape_detector.py.py' of the project from the project https://github.com/0-ma/geometric-shape-detector. No external code sources were used.
|