Update README.md
Browse files
README.md
CHANGED
@@ -42,4 +42,8 @@ pipeline_tag: image-classification
|
|
42 |
logits = model(**inputs)['logits'].cpu().detach().numpy()
|
43 |
predictions = np.argmax(logits, axis=1)
|
44 |
predicted_labels = [labels[prediction] for prediction in predictions]
|
45 |
-
print(predicted_labels)
|
|
|
|
|
|
|
|
|
|
42 |
logits = model(**inputs)['logits'].cpu().detach().numpy()
|
43 |
predictions = np.argmax(logits, axis=1)
|
44 |
predicted_labels = [labels[prediction] for prediction in predictions]
|
45 |
+
print(predicted_labels)
|
46 |
+
|
47 |
+
### License
|
48 |
+
|
49 |
+
The license for this model can be found [here](https://github.com/NVlabs/SegFormer/blob/master/LICENSE).
|