Skip to content

anamitradebnath/Image-Classification

Repository files navigation

App: Image Identification and Classification using Zero-shot

  • An app to use zero-shot image classification models to analyze and describe an image and identify and classify animals in it.
  • All these while creating custom AI Agent.

Demo

Screenshot 2024-12-01 at 7 00 16 PM Screenshot 2024-12-01 at 7 01 19 PM

How to Use

  • npm install
  • npm run dev

Todo

  • [Done] use zero-shot model, facebook-bart-large-mnli
  • [Done] use state management like recoil for smoother state management
  • Fix issue while calling classification and Summarize

Technologies used

  • vision/image model: nlpconnect/vit-gpt2-image-captioning
  • React
  • Typescript
  • Recoil
  • Zero-shot classification model
  • Custom AI Agent