AIRider commited on
Commit
75283fa
1 Parent(s): 43764c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -100,7 +100,7 @@ def clean_text(text):
100
  def create_client(model_name):
101
  return InferenceClient(model_name, token=os.getenv("HF_TOKEN"))
102
 
103
- client = create_client("OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5")
104
 
105
  def call_api(content, system_message, max_tokens, temperature, top_p):
106
  messages = [{"role": "system", "content": system_message}, {"role": "user", "content": content}]
 
100
  def create_client(model_name):
101
  return InferenceClient(model_name, token=os.getenv("HF_TOKEN"))
102
 
103
+ client = create_client("CohereForAI/c4ai-command-r-plus")
104
 
105
  def call_api(content, system_message, max_tokens, temperature, top_p):
106
  messages = [{"role": "system", "content": system_message}, {"role": "user", "content": content}]