Johnny commited on
Commit
5075a68
·
1 Parent(s): 7cf7e72

update to bart model api

Browse files
Files changed (1) hide show
  1. config.py +1 -1
config.py CHANGED
@@ -19,7 +19,7 @@ supabase = create_client(SUPABASE_URL, SUPABASE_KEY)
19
  embedding_model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
20
 
21
  HF_MODELS = {
22
- "bart": "https://router.huggingface.co/hf-inference/models/transformer3/H2-keywordextractor"
23
  }
24
 
25
  HF_API_TOKEN = os.getenv("HF_API_TOKEN")
 
19
  embedding_model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
20
 
21
  HF_MODELS = {
22
+ "bart": "https://router.huggingface.co/hf-inference/models/facebook/bart-large-cnn"
23
  }
24
 
25
  HF_API_TOKEN = os.getenv("HF_API_TOKEN")