JustKiddo commited on
Commit
4837ac0
·
verified ·
1 Parent(s): 736b7db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -5,7 +5,10 @@ from datasets import load_dataset
5
  """
6
  For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
7
  """
8
- client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
 
 
9
  #topic_model = BERTopic.load("MaartenGr/BERTopic_Wikipedia")
10
  # Train model
11
  #topic_model = BERTopic("english")
 
5
  """
6
  For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
7
  """
8
+
9
+ #Update: Using a new base model
10
+ #client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
11
+ client = InferenceClient("01-ai/Yi-1.5-34B-Chat")
12
  #topic_model = BERTopic.load("MaartenGr/BERTopic_Wikipedia")
13
  # Train model
14
  #topic_model = BERTopic("english")