This repository focuses on developing models for the machine learning part of TechWas. We experiment with transfer learning, transformers, and other techniques to find the best output for our tasks.
Our project focuses on two main tasks:
- Image classification: Our goal is to accurately classify images into 15 predefined categories.
- Text summarization: Our goal is to generate concise and informative summaries of longer text of article.
To achieve our goals, we use transfer learning techniques to leverage pre-trained models and fine-tune them with our own dataset for our specific tasks. This allows us to quickly develop high-performing models without the need for extensive training data or computational resources.
You can preview the implementation in this web app!
- text classification model (optional)
- object detection model (optional)
-
Image Classification
-
Base model
-
Fine tuned
-
-
Text Summarization
