Shankarm08 commited on
Commit
4c7c435
1 Parent(s): 60c0710

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -1,5 +1,3 @@
1
- # Load model directly
2
- from transformers import AutoImageProcessor, AutoModelForImageClassification
3
 
4
- processor = AutoImageProcessor.from_pretrained("tonyassi/fashion-clothing-decade")
5
- model = AutoModelForImageClassification.from_pretrained("tonyassi/fashion-clothing-decade")
 
1
+ from transformers import pipeline
 
2
 
3
+ pipe = pipeline("image-classification", model="tonyassi/fashion-clothing-decade")