zamasam commited on
Commit
8ca4c25
1 Parent(s): 26c234b

Update constants.py

Browse files
Files changed (1) hide show
  1. constants.py +1 -1
constants.py CHANGED
@@ -3,7 +3,7 @@ DEFAULT_CUDA_DEVICE = "cuda:0"
3
  # Also try: 'Qiliang/bart-large-cnn-samsum-ElectrifAi_v10'
4
  DEFAULT_SUMMARIZATION_MODEL = "Qiliang/bart-large-cnn-samsum-ChatGPT_v3"
5
  # Also try: 'joeddav/distilbert-base-uncased-go-emotions-student'
6
- DEFAULT_CLASSIFICATION_MODEL = "nateraw/bert-base-uncased-emotion"
7
  # Also try: 'Salesforce/blip-image-captioning-base'
8
  DEFAULT_CAPTIONING_MODEL = "Salesforce/blip-image-captioning-large"
9
  DEFAULT_SD_MODEL = "ckpt/anything-v4.5-vae-swapped"
 
3
  # Also try: 'Qiliang/bart-large-cnn-samsum-ElectrifAi_v10'
4
  DEFAULT_SUMMARIZATION_MODEL = "Qiliang/bart-large-cnn-samsum-ChatGPT_v3"
5
  # Also try: 'joeddav/distilbert-base-uncased-go-emotions-student'
6
+ DEFAULT_CLASSIFICATION_MODEL = "joeddav/distilbert-base-uncased-go-emotions-student"
7
  # Also try: 'Salesforce/blip-image-captioning-base'
8
  DEFAULT_CAPTIONING_MODEL = "Salesforce/blip-image-captioning-large"
9
  DEFAULT_SD_MODEL = "ckpt/anything-v4.5-vae-swapped"