Spaces:

data-agents
/

jupyter-agent

Running

lvwerra HF Staff commited on 13 days ago

Commit

0f4c888

verified ·

1 Parent(s): 4edb903

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -49,7 +49,7 @@ def execute_jupyter_agent(
     os.makedirs(save_dir, exist_ok=True)
     save_dir = os.path.join(save_dir, 'jupyter-agent.ipynb')
-    client = InferenceClient(api_key=HF_TOKEN, provider="auto")
     tokenizer = AutoTokenizer.from_pretrained(model)
     # model = "meta-llama/Llama-3.1-8B-Instruct"

     os.makedirs(save_dir, exist_ok=True)
     save_dir = os.path.join(save_dir, 'jupyter-agent.ipynb')
+    client = InferenceClient(api_key=HF_TOKEN, provider="hf-inference")
     tokenizer = AutoTokenizer.from_pretrained(model)
     # model = "meta-llama/Llama-3.1-8B-Instruct"