πŸ›οΈ 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