Restaurant-Token-Classifier

DistilBERT for Token Classification

📝 Overview

This model is a fine-tuned version of DistilBERT for token classification on restaurant-related data. It is capable of recognizing various entities such as:

• Dishes
• Amenities
• Ratings
• Restaurant names
• Opening hours
• Locations
• Prices
• Cuisine types

🧠 Model Details

• Architecture: DistilBertForTokenClassification
• Hidden Size: 768
• FFN Inner Hidden Size: 3072
• Attention Heads: 12
• Layers: 6
• Max Position Embeddings: 512
• Dropout: 0.1
• Tokenizer: DistilBERT Tokenizer
• Transformers Version: 4.51.3

🏷️ Labels

The model supports the following 17 token classification labels:

🚀 Usage Example

from transformers import DistilBertTokenizer, DistilBertForTokenClassification

# Load tokenizer and model
tokenizer = DistilBertTokenizer.from_pretrained("AbhishekBhavnani/Restaurant-Token-Classifier")
model = DistilBertForTokenClassification.from_pretrained("AbhishekBhavnani/Restaurant-Token-Classifier")

# Example input
inputs = tokenizer("Find The Best Place To Eat Pizza in Ahmedabad", return_tensors="pt")
outputs = model(**inputs)

# Get predictions
predictions = outputs.logits.argmax(dim=-1)
print(predictions)