Hugging Face's logo

infinity6
/
ecomm_shop_intent_pretrained

Transformers
Safetensors
ecomm_shop_intent_pretrained
recommendation
e-commerce
sequential-modeling
next-item-prediction
behavioral-modeling
Model card Files
xet
 Community
i6-aiworks commited on
Commit
ca057d3
·
verified ·
1 Parent(s): f0ab69f

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. sample.py +1 -1
README.md CHANGED
@@ -91,7 +91,7 @@ import torch
91
  from i6modelecomm.model import i6modelecomm
92
 
93
  model = i6modelecomm.CommerceIntent.from_pretrained(
94
- "i6-aiworks/ecomm_shop_intent_pretrained"
95
  )
96
 
97
  # TODO: map items and remap categories.
 
91
  from i6modelecomm.model import i6modelecomm
92
 
93
  model = i6modelecomm.CommerceIntent.from_pretrained(
94
+ "infinity6/ecomm_shop_intent_pretrained"
95
  )
96
 
97
  # TODO: map items and remap categories.
sample.py CHANGED
@@ -3,7 +3,7 @@ import torch
3
  from i6modelecomm.model import i6modelecomm
4
 
5
  model = i6modelecomm.CommerceIntent.from_pretrained(
6
- "i6-aiworks/ecomm_shop_intent_pretrained"
7
  )
8
 
9
  # TODO: map items and remap categories.
 
3
  from i6modelecomm.model import i6modelecomm
4
 
5
  model = i6modelecomm.CommerceIntent.from_pretrained(
6
+ "infinity6/ecomm_shop_intent_pretrained"
7
  )
8
 
9
  # TODO: map items and remap categories.