boltuix / bert-local

🌍 bert-local — Your Smarter Nearby Assistant! 🗺️

Understand Intent, Find Nearby Solutions 💡
bert-local is an intelligent AI assistant powered by bert-mini , designed to interpret natural, conversational queries and suggest precise local business categories in real time. Unlike traditional map services that struggle with NLP, bert-local captures personal intent to deliver actionable results—whether it’s finding a 🐾 pet store for a sick dog or a 💼 accounting firm for tax help.

With support for 140+ local business categories and a compact model size of ~20MB , bert-local combines open-source datasets and advanced fine-tuning to overcome the limitations of Google Maps’ NLP. Open source and extensible, it’s perfect for developers and businesses building context-aware local search solutions on edge devices and mobile applications. 🚀

Explore bert-local 🌟

Table of Contents 📋

• Why bert-local? 🌈
• Key Features ✨
• Supported Categories 🏪
• Installation 🛠️
• Quickstart: Dive In 🚀
• Training the Model 🧠
• Evaluation 📈
• Dataset Details 📊
• Use Cases 🌍
• Comparison to Other Solutions ⚖️
• Source 🌱
• License 📜
• Credits 🙌
• Community & Support 🌐
• Last Updated 📅

---

Why bert-local? 🌈

• Intent-Driven 🧠: Understands natural language queries like “My dog isn’t eating” to suggest 🐾 pet stores or 🩺 veterinary clinics.
• Accurate & Fast ⚡: Achieves 94.26% test accuracy (115/122 correct) for precise category predictions in real time.
• Extensible 🛠️: Open source and customizable with your own datasets (e.g., ChatGPT, Grok, or proprietary data).
• Comprehensive 🏪: Supports 140+ local business categories , from 💼 accounting firms to 🦒 zoos.
• Lightweight 📱: Compact ~20MB model size, optimized for edge devices and mobile applications.

“bert-local transformed our app’s local search—it feels like it gets the user!” — App Developer 💬

---

Key Features ✨

• Advanced NLP 📜: Built on bert-mini , fine-tuned for multi-class text classification.
• Real-Time Results ⏱️: Delivers category suggestions instantly, even for complex queries.
• Wide Coverage 🗺️: Matches queries to 140+ business categories with high confidence.
• Developer-Friendly 🧑‍💻: Easy integration with Python 🐍, Hugging Face 🤗, and custom APIs.
• Open Source 🌐: Freely extend and adapt for your needs.

---

🔧 How to Use

from transformers import pipeline  # 🤗 Import Hugging Face pipeline

# 🚀 Load the fine-tuned intent classification model
classifier = pipeline("text-classification", model="boltuix/bert-local")

# 🧠 Predict the user's intent from a sample input sentence
result = classifier("Where can I see ocean creatures behind glass?")  # 🐠 Expecting Aquarium

# 📊 Print the classification result with label and confidence score
print(result)  # 🖨️ Example output: [{'label': 'aquarium', 'score': 0.999}]

---

Supported Categories 🏪

bert-local supports 140 local business categories , each paired with an emoji for clarity:

• 💼 Accounting Firm
• ✈️ Airport
• 🎢 Amusement Park
• 🐠 Aquarium
• 🖼️ Art Gallery
• 🏧 ATM
• 🚗 Auto Dealership
• 🔧 Auto Repair Shop
• 🥐 Bakery
• 🏦 Bank
• 🍻 Bar
• 💈 Barber Shop
• 🏖️ Beach
• 🚲 Bicycle Store
• 📚 Book Store
• 🎳 Bowling Alley
• 🚌 Bus Station
• 🥩 Butcher Shop
• ☕ Cafe
• 📸 Camera Store
• ⛺ Campground
• 🚘 Car Rental
• 🧼 Car Wash
• 🎰 Casino
• ⚰️ Cemetery
• ⛪ Church
• 🏛️ City Hall
• 🩺 Clinic
• 👗 Clothing Store
• ☕ Coffee Shop
• 🏪 Convenience Store
• 🍳 Cooking School
• 🖨️ Copy Center
• 📦 Courier Service
• ⚖️ Courthouse
• ✂️ Craft Store
• 💃 Dance Studio
• 🦷 Dentist
• 🏬 Department Store
• 🩺 Doctor’s Office
• 💊 Drugstore
• 🧼 Dry Cleaner
• ⚡️ Electrician
• 📱 Electronics Store
• 🏫 Elementary School
• 🏛️ Embassy
• 🚒 Fire Station
• 💐 Florist
• 🎮 Gaming Center
• ⚰️ Funeral Home
• 🎁 Gift Shop
• 🌸 Flower Shop
• 🔩 Hardware Store
• 💇 Hair Salon
• 🔨 Handyman
• 🧹 House Cleaning
• 🛠️ House Painter
• 🏠 Home Goods Store
• 🏥 Hospital
• 🕉️ Hindu Temple
• 🌳 Gardening Service
• 🏡 Lodging
• 🔒 Locksmith
• 🧼 Laundromat
• 📚 Library
• 🚈 Light Rail Station
• 🛡️ Insurance Agency
• ☕ Internet Cafe
• 🏨 Hotel
• 💎 Jewelry Store
• 🗣️ Language School
• 🛍️ Market
• 🍽️ Meal Delivery Service
• 🕌 Mosque
• 🎥 Movie Theater
• 🚚 Moving Company
• 🏛️ Museum
• 🎵 Music School
• 🎸 Music Store
• 💅 Nail Salon
• 🎉 Night Club
• 🌱 Nursery
• 🖌️ Office Supply Store
• 🌳 Park
• 🚗 Parking Lot
• 🐜 Pest Control Service
• 🐾 Pet Grooming
• 🐶 Pet Store
• 💊 Pharmacy
• 📷 Photography Studio
• 🩺 Physiotherapist
• 💉 Piercing Shop
• 🚰 Plumbing Service
• 🚓 Police Station
• 📚 Public Library
• 🚻 Public Restroom
• 🏠 Real Estate Agency
• ♻️ Recycling Center
• 🍽️ Restaurant
• 🏠 Roofing Contractor
• 🏫 School
• 📦 Shipping Center
• 👞 Shoe Store
• 🏬 Shopping Mall
• ⛸️ Skating Rink
• ❄️ Snow Removal Service
• 🧘 Spa
• 🏀 Sport Store
• 🏟️ Stadium
• 📜 Stationary Store
• 📦 Storage Facility
• 🚇 Subway Station
• 🛒 Supermarket
• 🕍 Synagogue
• ✂️ Tailor
• 🎨 Tattoo Parlor
• 🚕 Taxi Stand
• 🚗 Tire Shop
• 🗺️ Tourist Attraction
• 🧸 Toy Store
• 🎲 Toy Lending Library
• 🚂 Train Station
• 🚆 Transit Station
• ✈️ Travel Agency
• 🏫 University
• 📼 Video Rental Store
• 🍷 Wine Shop
• 🧘 Yoga Studio
• 🦒 Zoo
• ⛽ Gas Station
• 📯 Post Office
• 💪 Gym
• 🏘️ Community Center
• 🏪 Grocery Store

---

Installation 🛠️

Get started with bert-local:

pip install transformers torch pandas scikit-learn tqdm

• Requirements 📋: Python 3.8+, ~20MB storage for model and dependencies.
• Optional 🔧: CUDA-enabled GPU for faster training/inference.
• Model Download 📥: Grab the pre-trained model from Hugging Face .

---

Quickstart: Dive In 🚀

from transformers import AutoModelForSequenceClassification

# 📥 Load the fine-tuned intent classification model
model = AutoModelForSequenceClassification.from_pretrained("boltuix/bert-local")

# 🏷️ Extract the ID-to-label mapping dictionary
label_mapping = model.config.id2label

# 📋 Convert and sort all labels to a clean list
supported_labels = sorted(label_mapping.values())

# ✅ Print the supported categories
print("✅ Supported Categories:", supported_labels)

---

Training the Model 🧠

bert-local is trained using bert-mini for multi-class text classification. Here’s how to train it:

Prerequisites

• Dataset in CSV format with text (query) and label (category) columns.
• Example dataset structure:

  text,label
  "Need help with taxes","accounting firm"
  "Where’s the nearest airport?","airport"
  ...
  

Training Code

• 📍 Get training Source Code 🌟
• 📍 Dataset (comming soon..)

---

Evaluation 📈

bert-local was tested on 122 test cases , achieving 94.26% accuracy (115/122 correct). Below are sample results:

Evaluation Metrics

Note : F1 score is estimated based on high accuracy. Test with your dataset for precise metrics.

---

Dataset Details 📊

• Source : Open-source datasets, augmented with custom queries (e.g., ChatGPT, Grok, or proprietary data).
• Format : CSV with text (query) and label (category) columns.
• Categories : 140 (see Supported Categories ).
• Size : Varies based on dataset; model footprint ~20MB.
• Preprocessing : Handled via tokenization and label encoding (see Training the Model ).

---

Use Cases 🌍

bert-local powers a variety of applications:

• Local Search Apps 🗺️: Suggest 🐾 pet stores or 🩺 clinics based on queries like “My dog is sick.”
• Chatbots 🤖: Enhance customer service bots with context-aware local recommendations.
• E-Commerce 🛍️: Guide users to nearby 💼 accounting firms or 📚 bookstores.
• Travel Apps ✈️: Recommend 🏨 hotels or 🗺️ tourist attractions for travelers.
• Healthcare 🩺: Direct users to 🏥 hospitals or 💊 pharmacies for urgent needs.
• Smart Assistants 📱: Integrate with voice assistants for hands-free local search.

---

Comparison to Other Solutions ⚖️

bert-local excels with its high accuracy , strong NLP , and open-source flexibility . 🚀

---

Source 🌱

• Base Model : bert-mini.
• Data : Open-source datasets, synthetic queries, and community contributions.
• Mission : Make local search intuitive and intent-driven for all.

---

License 📜

Open Source : Free to use, modify, and distribute under Apache-2.0. See repository for details.

---

Credits 🙌

• Developed By : [bert-local team] 👨‍💻
• Base Model : bert-mini 🧠
• Powered By : Hugging Face 🤗, PyTorch 🔥, and open-source datasets 🌐

---

Community & Support 🌐

Join the bert-local community:

• 📍 Explore the Hugging Face model page 🌟
• 🛠️ Report issues or contribute at the repository 🔧
• 💬 Discuss on Hugging Face forums or submit pull requests 🗣️
• 📚 Learn more via Hugging Face Transformers docs 📖

Your feedback shapes bert-local! 😊

---

Last Updated 📅

June 9, 2025 — Added 140+ category support, updated test accuracy, and enhanced documentation with emojis.

Get Started with bert-local 🚀