ποΈ EasySell AI β E-Commerce Chatbot SLM
A 100M parameter GPT-style Small Language Model trained for e-commerce intent classification and tool-calling agent for online shopping.
Model Details
- Architecture: Custom GPT (14 layers, 12 heads, 768 dim)
- Parameters: ~100M
- Trained on: 30,000 synthetic e-commerce conversations
- Intents: GET_LINK, SEARCH_PRODUCT, ADD_TO_CART, CHECKOUT, FAQ, CLARIFY, REFUSE
Features
- π Add to cart with quantity control
- π¦ Order tracking, cancel & refund workflows
- π§Ύ Invoice generation with GST
- π³ Razorpay payment integration
- ποΈ Coupon system
- π Product comparison
- π€ 50+ FAQ auto-answers
- π¨βπΌ Human agent routing
Usage
import torch, json
from huggingface_hub import hf_hub_download
# Download model
model_path = hf_hub_download(repo_id="YOUR_USERNAME/easysell-ai", filename="model.pt")
config_path = hf_hub_download(repo_id="YOUR_USERNAME/easysell-ai", filename="config.json")
# Load config
with open(config_path) as f:
cfg = json.load(f)
# Load model (after defining your GPT class)
model.load_state_dict(torch.load(model_path, map_location="cpu"))
model.eval()
Demo
Run the Gradio UI using GRADIO_FINAL.py after loading the model.
Files
| File | Description |
|---|---|
model.pt |
Model weights |
config.json |
Model architecture config |
tokenizer.json |
BPE tokenizer |
products_description.json |
Product catalog |
products_images.json |
Product image URLs |
CELL_AGENT_FINAL.py |
Tool-calling agent |
GRADIO_FINAL.py |
Gradio chat UI |
- Downloads last month
- 105
Inference Providers NEW
This model isn't deployed by any Inference Provider. π Ask for provider support