
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
should probably proofread and complete it, then remove this comment. -->

# Celebrity Classifier
## Model description
This model classifies a face to a celebrity. It is trained on [ares1123/celebrity_dataset](https://huggingface.co/datasets/ares1123/celebrity_dataset) dataset and fine-tuned on [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k).

## Dataset description
[ares1123/celebrity_dataset](https://huggingface.co/datasets/ares1123/celebrity_dataset)
Top 1000 celebrities. 18,184 images. 256x256. Square cropped to face.

### How to use
```python
from transformers import pipeline

# Initialize image classification pipeline
pipe = pipeline("image-classification", model="tonyassi/celebrity-classifier")

# Perform classification
result = pipe('image.png')

# Print results
print(result)
```

## Training and evaluation data
It achieves the following results on the evaluation set:
- Loss: 0.9089
- Accuracy: 0.7982

### Training hyperparameters

The following hyperparameters were used during training:
- learning_rate: 5e-05
- train_batch_size: 16
- eval_batch_size: 16
- seed: 42
- gradient_accumulation_steps: 4
- total_train_batch_size: 64
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: linear
- lr_scheduler_warmup_ratio: 0.1
- num_epochs: 20

### Framework versions

- Transformers 4.35.2
- Pytorch 2.1.0+cu121
- Datasets 2.16.1
- Tokenizers 0.15.0

aaron eckhart aaron paul adam driver blake lively bob odenkirk bonnie wright boyd holbrook brad pitt bradley cooper brendan fraser brian cox brie larson brittany snow adam lambert bryan cranston bryce dallas howard busy philipps caitriona balfe cameron diaz camila cabello camila mendes cardi b carey mulligan carla gugino adam levine carrie underwood casey affleck cate blanchett catherine keener catherine zeta-jones celine dion chace crawford chadwick boseman channing tatum charlie cox adam sandler charlie day charlie hunnam charlie plummer charlize theron chiara ferragni chiwetel ejiofor chloe bennet chloe grace moretz chloe sevigny chloë grace moretz adam scott chloë sevigny chris cooper chris evans chris hemsworth chris martin chris messina chris noth chris o'dowd chris pine chris pratt adele chris tucker chrissy teigen christian bale christian slater christina aguilera christina applegate christina hendricks christina milian christina ricci christine baranski adrian grenier christoph waltz christopher plummer christopher walken cillian murphy claire foy clive owen clive standen cobie smulders colin farrell colin firth adèle exarchopoulos colin hanks connie britton conor mcgregor constance wu constance zimmer courteney cox cristiano ronaldo daisy ridley dak prescott dakota fanning aidan gillen dakota johnson damian lewis dan stevens danai gurira dane dehaan daniel craig daniel dae kim daniel day-lewis daniel gillies daniel kaluuya aidan turner daniel mays daniel radcliffe danny devito darren criss dave bautista dave franco dave grohl daveed diggs david attenborough david beckham aaron rodgers aishwarya rai david duchovny david harbour david oyelowo david schwimmer david tennant david thewlis dax shepard debra messing demi lovato dennis quaid aja naomi king denzel washington dermot mulroney dev patel diane keaton diane kruger diane lane diego boneta diego luna djimon hounsou dolly parton alden ehrenreich domhnall gleeson dominic cooper dominic monaghan dominic west don cheadle donald glover donald sutherland donald trump dua lipa dwayne "the rock" johnson aldis hodge dwayne johnson dylan o'brien ed harris ed helms ed sheeran eddie murphy eddie redmayne edgar ramirez edward norton eiza gonzalez alec baldwin eiza gonzález elijah wood elisabeth moss elisha cuthbert eliza coupe elizabeth banks elizabeth debicki elizabeth lail elizabeth mcgovern elizabeth moss alex morgan elizabeth olsen elle fanning ellen degeneres ellen page ellen pompeo ellie goulding elon musk emile hirsch emilia clarke emilia fox alex pettyfer emily beecham emily blunt emily browning emily deschanel emily hampshire emily mortimer emily ratajkowski emily vancamp emily watson emma bunton alex rodriguez emma chamberlain emma corrin emma mackey emma roberts emma stone emma thompson emma watson emmanuelle chriqui emmy rossum eoin macken alexander skarsgård eric bana ethan hawke eva green eva longoria eva mendes evan peters evan rachel wood evangeline lilly ewan mcgregor ezra miller alexandra daddario felicity huffman felicity jones finn wolfhard florence pugh florence welch forest whitaker freddie highmore freddie prinze jr. freema agyeman freida pinto aaron taylor-johnson alfre woodard freya allan gabrielle union gael garcia bernal gael garcía bernal gal gadot garrett hedlund gary oldman gemma arterton gemma chan gemma whelan alia shawkat george clooney george lucas gerard butler giancarlo esposito giannis antetokounmpo gigi hadid gillian anderson gillian jacobs gina carano gina gershon alice braga gina rodriguez ginnifer goodwin gisele bundchen glenn close grace kelly greg kinnear greta gerwig greta scacchi greta thunberg gugu mbatha-raw alice eve guy ritchie gwen stefani gwendoline christie gwyneth paltrow hafthor bjornsson hailee steinfeld hailey bieber haley joel osment halle berry hannah simone alicia keys harrison ford harry styles harvey weinstein hayden panettiere hayley atwell helen hunt helen mirren helena bonham carter henry cavill henry golding alicia vikander hilary swank himesh patel hozier hugh bonneville hugh dancy hugh grant hugh jackman hugh laurie ian somerhalder idris elba alison brie imelda staunton imogen poots ioan gruffudd isabella rossellini isabelle huppert isla fisher issa rae iwan rheon j.k. rowling j.k. simmons allison janney jack black jack reynor jack whitehall jackie chan jada pinkett smith jaden smith jaimie alexander jake gyllenhaal jake johnson jake t. austin allison williams james cameron james corden james franco james marsden james mcavoy james norton jamie bell jamie chung jamie dornan jamie foxx alyson hannigan jamie lee curtis jamie oliver jane fonda jane krakowski jane levy jane lynch jane seymour janelle monáe january jones jared leto abbi jacobson amanda peet jason bateman jason clarke jason derulo jason isaacs jason momoa jason mraz jason schwartzman jason segel jason statham jason sudeikis amanda seyfried javier bardem jay baruchel jay-z jeff bezos jeff bridges jeff daniels jeff goldblum jeffrey dean morgan jeffrey donovan jeffrey wright amandla stenberg jemima kirke jenna coleman jenna fischer jenna ortega jennifer aniston jennifer connelly jennifer coolidge jennifer esposito jennifer garner jennifer hudson amber heard jennifer lawrence jennifer lopez jennifer love hewitt jenny slate jeremy irons jeremy renner jeremy strong jerry seinfeld jesse eisenberg jesse metcalfe america ferrera jesse plemons jesse tyler ferguson jesse williams jessica alba jessica biel jessica chastain jessica lange jessie buckley jim carrey jim parsons amy adams joan collins joan cusack joanne froggatt joaquin phoenix jodie comer jodie foster joe jonas joe keery joel edgerton joel kinnaman amy poehler joel mchale john boyega john c. reilly john cena john cho john cleese john corbett john david washington john goodman john hawkes amy schumer john krasinski john legend john leguizamo john lithgow john malkovich john mayer john mulaney john oliver john slattery john travolta ana de armas john turturro johnny depp johnny knoxville jon bernthal jon favreau jon hamm jonah hill jonathan groff jonathan majors jonathan pryce andie macdowell jonathan rhys meyers jordan peele jordana brewster joseph fiennes joseph gordon-levitt josh allen josh brolin josh gad josh hartnett josh hutcherson abhishek bachchan andrew garfield josh radnor jude law judy dench judy greer julia garner julia louis-dreyfus julia roberts julia stiles julian casablancas julian mcmahon andrew lincoln julianna margulies julianne hough julianne moore julianne nicholson juliette binoche juliette lewis juno temple jurnee smollett justin bartha justin bieber andrew scott justin hartley justin herbert justin long justin theroux justin timberlake kj apa kaitlyn dever kaley cuoco kanye west karl urban andy garcia kat dennings kate beckinsale kate bosworth kate hudson kate mara kate middleton kate upton kate walsh kate winslet katee sackhoff andy samberg katherine heigl katherine langford katherine waterston kathryn hahn katie holmes katie mcgrath katy perry kaya scodelario keanu reeves keegan-michael key andy serkis keira knightley keke palmer kelly clarkson kelly macdonald kelly marie tran kelly reilly kelly ripa kelvin harrison jr. keri russell kerry washington angela bassett kevin bacon kevin costner kevin hart kevin spacey ki hong lee kiefer sutherland kieran culkin kiernan shipka kim dickens kim kardashian angelina jolie kirsten dunst kit harington kourtney kardashian kristen bell kristen stewart kristen wiig kristin davis krysten ritter kyle chandler kylie jenner anna camp kylie minogue lady gaga lake bell lakeith stanfield lamar jackson lana del rey laura dern laura harrier laura linney laura prepon anna faris laurence fishburne laverne cox lebron james lea michele lea seydoux lee pace leighton meester lena headey leonardo da vinci leonardo dicaprio abigail breslin anna kendrick leslie mann leslie odom jr. lewis hamilton liam hemsworth liam neeson lili reinhart lily aldridge lily allen lily collins lily james anna paquin lily rabe lily tomlin lin-manuel miranda linda cardellini lionel messi lisa bonet lisa kudrow liv tyler lizzo logan lerman annasophia robb lorde lucy boynton lucy hale lucy lawless lucy liu luke evans luke perry luke wilson lupita nyong'o léa seydoux annabelle wallis mackenzie davis madelaine petsch mads mikkelsen mae whitman maggie gyllenhaal maggie q maggie siff maggie smith mahershala ali mahira khan anne hathaway maisie richardson-sellers maisie williams mandy moore mandy patinkin marc anthony margaret qualley margot robbie maria sharapova marion cotillard marisa tomei anne marie mariska hargitay mark hamill mark ruffalo mark strong mark wahlberg mark zuckerberg marlon brando martin freeman martin scorsese mary elizabeth winstead anne-marie mary j. blige mary steenburgen mary-louise parker matt bomer matt damon matt leblanc matt smith matthew fox matthew goode matthew macfadyen ansel elgort matthew mcconaughey matthew perry matthew rhys matthew stafford max minghella maya angelou maya hawke maya rudolph megan fox megan rapinoe anson mount meghan markle mel gibson melanie lynskey melissa benoist melissa mccarthy melonie diaz meryl streep mia wasikowska michael b. jordan michael c. hall anthony hopkins michael caine michael cera michael cudlitz michael douglas michael ealy michael fassbender michael jordan michael keaton michael pena michael peña abigail spencer anthony joshua michael phelps michael shannon michael sheen michael stuhlbarg michelle dockery michelle monaghan michelle obama michelle pfeiffer michelle rodriguez michelle williams anthony mackie michelle yeoh michiel huisman mila kunis miles teller milla jovovich millie bobby brown milo ventimiglia mindy kaling miranda cosgrove miranda kerr antonio banderas mireille enos molly ringwald morgan freeman mélanie laurent naomi campbell naomi harris naomi scott naomi watts naomie harris nas anya taylor-joy natalie dormer natalie imbruglia natalie morales natalie portman nathalie emmanuel nathalie portman nathan fillion naya rivera neil patrick harris neil degrasse tyson ariana grande neve campbell neymar jr. nicholas braun nicholas hoult nick jonas nick kroll nick offerman nick robinson nicole kidman nikolaj coster-waldau armie hammer nina dobrev noah centineo noomi rapace norman reedus novak djokovic octavia spencer odessa young odette annable olivia colman olivia cooke ashley judd olivia holt olivia munn olivia wilde oprah winfrey orlando bloom oscar isaac owen wilson pablo picasso patrick dempsey patrick mahomes ashton kutcher patrick stewart patrick wilson paul bettany paul dano paul giamatti paul mccartney paul rudd paul wesley paula patton pedro almodóvar aubrey plaza pedro pascal penelope cruz penélope cruz pete davidson peter dinklage phoebe dynevor phoebe waller-bridge pierce brosnan portia de rossi priyanka chopra auli'i cravalho quentin tarantino rachel bilson rachel brosnahan rachel mcadams rachel weisz rafe spall rainn wilson ralph fiennes rami malek rashida jones adam brody awkwafina ray liotta ray romano rebecca ferguson rebecca hall reese witherspoon regina hall regina king renee zellweger renée zellweger rhys ifans barack obama ricardo montalban richard armitage richard gere richard jenkins richard madden ricky gervais ricky martin rihanna riley keough rita ora bella hadid river phoenix riz ahmed rob lowe robert carlyle robert de niro robert downey jr. robert pattinson robert sheehan robin tunney robin williams bella thorne roger federer rooney mara rosamund pike rosario dawson rose byrne rose leslie roselyn sanchez ruby rose rupert grint russell brand ben barnes russell crowe russell wilson ruth bader ginsburg ruth wilson ryan eggold ryan gosling ryan murphy ryan phillippe ryan reynolds ryan seacrest ben mendelsohn salma hayek sam claflin sam heughan sam rockwell sam smith samara weaving samuel l. jackson sandra bullock sandra oh saoirse ronan ben stiller sarah gadon sarah hyland sarah jessica parker sarah michelle gellar sarah paulson sarah silverman sarah wayne callies sasha alexander scarlett johansson scott speedman ben whishaw sean bean sebastian stan selena gomez selma blair serena williams seth macfarlane seth meyers seth rogen shailene woodley shakira benedict cumberbatch shania twain sharlto copley shawn mendes shia labeouf shiri appleby shohreh aghdashloo shonda rhimes sienna miller sigourney weaver simon baker benedict wong simon cowell simon pegg simone biles sofia boutella sofia vergara sophie turner sophie wessex stanley tucci stephen amell stephen colbert adam devine benicio del toro stephen curry stephen dorff sterling k. brown sterling knight steve carell steven yeun susan sarandon taika waititi taraji p. henson taron egerton bill gates taylor hill taylor kitsch taylor lautner taylor schilling taylor swift teresa palmer terrence howard tessa thompson thandie newton the weeknd bill hader theo james thomas brodie-sangster thomas jane tiger woods tilda swinton tim burton tim cook timothee chalamet timothy olyphant timothy spall bill murray timothée chalamet tina fey tobey maguire toby jones toby kebbell toby regbo tom brady tom brokaw tom cavanagh tom cruise bill pullman tom ellis tom felton tom hanks tom hardy tom hiddleston tom holland tom hollander tom hopper tom selleck toni collette bill skarsgård tony hale topher grace tracee ellis ross tyra banks tyrese gibson uma thurman usain bolt uzo aduba vanessa hudgens vanessa kirby billie eilish vera farmiga victoria pedretti viggo mortensen vin diesel vince vaughn vincent cassel vincent d'onofrio vincent kartheiser viola davis walton goggins billie lourd wes anderson wes bentley whoopi goldberg will ferrell will poulter willem dafoe william jackson harper william shatner winona ryder woody harrelson billy crudup yara shahidi yvonne strahovski zac efron zach braff zach galifianakis zachary levi zachary quinto zayn malik zazie beetz zendaya billy porter zoe kazan zoe kravitz zoe saldana zoey deutch zooey deschanel zoë kravitz zoë saldana