Spaces:

supratipb
/

agent2

Sleeping

supratipb commited on Jul 11

Commit

5bbe004

verified ·

1 Parent(s): 9d57957

Upload app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,9 +20,9 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        #openAI_model = OpenAIServerModel(model_id="gpt-4o-mini", api_key="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")
         self.agent = CodeAgent(
-            model = InferenceClientModel(model_id="mistralai/Mistral-7B-Instruct-v0.1"),
             tools = [DuckDuckGoSearchTool(), WikipediaSearchTool()],
             add_base_tools = True,
         )

 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
+        openAI_model = OpenAIServerModel(model_id="gpt-4o-mini", api_key="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")
         self.agent = CodeAgent(
+            model = openAI_model,
             tools = [DuckDuckGoSearchTool(), WikipediaSearchTool()],
             add_base_tools = True,
         )