Spaces:

david-thrower
/

3B-Param-Agent

Sleeping

david-thrower commited on 9 days ago

Commit

8dcdef2

verified ·

1 Parent(s): d1ab25e

Update app.py

API version correction ...

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,6 @@ from smolagents import (
     FinalAnswerTool,
     CodeAgent,
     InferenceClientModel,
-    HfApiModel,
     PromptTemplates,
     GradioUI
 )
@@ -36,8 +35,7 @@ API_TOKEN_SMOLLM = getenv("HF_TOKEN")
 #         repetition_penalty=1.1
 #     )
-#model = InferenceClientModel(
-model = HfApiModel(
         model_id=MODEL_NAME,
         # max_new_tokens=32768, # ,  # 4096,
         token=API_TOKEN_SMOLLM,

     FinalAnswerTool,
     CodeAgent,
     InferenceClientModel,
     PromptTemplates,
     GradioUI
 )
 #         repetition_penalty=1.1
 #     )
+model = InferenceClientModel(
         model_id=MODEL_NAME,
         # max_new_tokens=32768, # ,  # 4096,
         token=API_TOKEN_SMOLLM,