22

utils/prompt_utils.py

@@ -22,11 +22,11 @@ def polish_prompt(original_prompt):
         return magic_prompt
     # client = InferenceClient(provider="cerebras", api_key=api_key)
     # messages = []
-    client = InferenceClient(provider="cerebras", api_key=api_key)
+    client = InferenceClient(provider="nebius", api_key=api_key)
     
     try:
         completion = client.chat.completions.create(
-            model="meta-llama/Llama-3.1-8B-Instruct",
+            model="Qwen/Qwen3-Coder-30B-A3B-Instruct",
             max_tokens=256,
             messages=[
                 {"role": "system", "content": system_prompt},