pmbrito commited on
Commit
9756209
·
1 Parent(s): 176067d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,4 +2,5 @@ from transformers import pipeline
2
 
3
  vision_classifier = pipeline(task="image-classification")
4
  result = vision_classifier(images="https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/pipeline-cat-chonk.jpeg")
5
- print("\n".join([f"Class {d['label']} with score {round(d['score'], 4)}" for d in result]))
 
 
2
 
3
  vision_classifier = pipeline(task="image-classification")
4
  result = vision_classifier(images="https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/pipeline-cat-chonk.jpeg")
5
+ print("\n".join([f"Class {d['label']} with score {round(d['score'], 4)}" for d in result]))
6
+ st.text("\n".join([f"Class {d['label']} with score {round(d['score'], 4)}" for d in result]))