from transformers import RobertaTokenizer, RobertaForSequenceClassification
import torch

model = RobertaForSequenceClassification.from_pretrained("YOUR_USERNAME/bugflow-team-classifier")
tokenizer = RobertaTokenizer.from_pretrained("YOUR_USERNAME/bugflow-team-classifier")

text = "Button not responding on click"
inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
outputs = model(**inputs)
probs = torch.softmax(outputs.logits, dim=1)
labels = ['Backend', 'Frontend', 'Mobile', 'DevOps']
predicted = labels[torch.argmax(probs).item()]
print(f"Team: {predicted}")

Training

Base model: microsoft/codebert-base
Dataset: Custom GitHub issues dataset + domain-specific bugs
Fine-tuned using Hugging Face Transformers

loke007/bugflow-team-classifier

BugFlow Team Classifier

Labels

Usage

Training

loke007/
bugflow-team-classifier