0-ma commited on
Commit
2f3e24c
1 Parent(s): dd3bfbd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -1
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).