YAML Metadata Warning:The pipeline tag "text2text-generation" is not in the official list: text-classification, token-classification, table-question-answering, question-answering, zero-shot-classification, translation, summarization, feature-extraction, text-generation, fill-mask, sentence-similarity, text-to-speech, text-to-audio, automatic-speech-recognition, audio-to-audio, audio-classification, audio-text-to-text, voice-activity-detection, depth-estimation, image-classification, object-detection, image-segmentation, text-to-image, image-to-text, image-to-image, image-to-video, unconditional-image-generation, video-classification, reinforcement-learning, robotics, tabular-classification, tabular-regression, tabular-to-text, table-to-text, multiple-choice, text-ranking, text-retrieval, time-series-forecasting, text-to-video, image-text-to-text, image-text-to-image, image-text-to-video, visual-question-answering, document-question-answering, zero-shot-image-classification, graph-ml, mask-generation, zero-shot-object-detection, text-to-3d, image-to-3d, image-feature-extraction, video-text-to-text, keypoint-detection, visual-document-retrieval, any-to-any, video-to-video, other

Vaishnavi ResumeBot - FLAN-T5

This repository contains a personal ResumeBot built by fine-tuning google/flan-t5-small on resume-based question-answer pairs.

About

The bot is designed to answer questions about Vaishnavi V M's academic background, technical skills, projects, and contact information.

Base Model

google/flan-t5-small

Main Topics

MSc Data Science at Dayananda Sagar University
BSc Mathematics and Computer Science
Python, Machine Learning, Deep Learning, NLP, SQL
Medical X-ray Abnormality Localization using Attention Mechanisms
Driver Drowsiness Detection
Phishing Website Detection
Language Identification and Accent Detection
Social Media Addiction Risk Prediction

Repository Files

Vaishnavi_ResumeBot_FLAN_T5.ipynb - complete Colab notebook
resume_qa_dataset.csv - training question-answer pairs
app.py - Gradio ResumeBot interface
inference.py - command-line prediction example
push_to_hub.py - upload your trained model to Hugging Face
requirements.txt - required libraries
.gitignore - files to ignore in Git
MODEL_FILES_README.txt - explains how trained model files are created

How to Train

Open Vaishnavi_ResumeBot_FLAN_T5.ipynb in Google Colab.

Use:

Runtime -> Change runtime type -> T4 GPU

Then run the notebook from top to bottom.

After training, save the model using:

model.save_pretrained("vaishnavi-resumebot-flan-t5")
tokenizer.save_pretrained("vaishnavi-resumebot-flan-t5")

This produces files such as:

config.json
generation_config.json
model.safetensors
tokenizer files
spiece.model

Run the App Locally

pip install -r requirements.txt
python app.py

Example Questions

What is your educational background?
What are your technical skills?
Tell me about your medical X-ray project.
Which algorithms did you use for phishing detection?
What is your language identification project?
What is your email address?

Important

The final fine-tuned model.safetensors is not included in the starter repository because it must be generated by actually training the model.

Downloads last month: -; Downloads are not tracked for this model. How to track

Model tree for vaishnavi-vm/Finetuning

Base model

google/flan-t5-small

Finetuned

(523)

this model