Alexander Bagus commited on
Commit
cf37355
·
1 Parent(s): ee3b958
Files changed (1) hide show
  1. utils/prompt_utils.py +1 -2
utils/prompt_utils.py CHANGED
@@ -14,11 +14,10 @@ def polish_prompt(original_prompt):
14
 
15
  # client = InferenceClient(provider="cerebras", api_key=api_key)
16
  # messages = []
17
- client = InferenceClient()
18
 
19
  try:
20
  completion = client.chat.completions.create(
21
- provider="cerebras",
22
  api_key=api_key,
23
  model="Qwen/Qwen3-235B-A22B-Instruct-2507",
24
  messages=[
 
14
 
15
  # client = InferenceClient(provider="cerebras", api_key=api_key)
16
  # messages = []
17
+ client = InferenceClient(provider="cerebras")
18
 
19
  try:
20
  completion = client.chat.completions.create(
 
21
  api_key=api_key,
22
  model="Qwen/Qwen3-235B-A22B-Instruct-2507",
23
  messages=[