Spaces:
Tiredmaker
/
Runtime error

Tiredmaker commited on
Commit
1cf9223
1 Parent(s): 570b0f3

Update constants.py

Browse files
Files changed (1) hide show
  1. constants.py +2 -2
constants.py CHANGED
@@ -1,9 +1,9 @@
1
  # Constants
2
  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"
 
1
  # Constants
2
  DEFAULT_CUDA_DEVICE = "cuda:0"
3
  # Also try: 'Qiliang/bart-large-cnn-samsum-ElectrifAi_v10'
4
+ DEFAULT_SUMMARIZATION_MODEL = "Qiliang/bart-large-cnn-samsum-ElectrifAi_v14"
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"