arpm01 commited on
Commit
c5287fe
·
1 Parent(s): b515163

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ model = AutoModelForImageClassification.from_pretrained("nateraw/vit-age-classif
7
 
8
  pipe = pipeline(task="image-classification",
9
  model=model,
10
- feature_extractor=extractor,)
11
 
12
  gr.Interface.load(pipe,
13
  title="age class",
 
7
 
8
  pipe = pipeline(task="image-classification",
9
  model=model,
10
+ )
11
 
12
  gr.Interface.load(pipe,
13
  title="age class",