Final_Assignment_Template

Sleeping

Mayur Shrivastav commited on Apr 12

Commit

af74eff

1 Parent(s): 5634c8c

updated requirement

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,8 @@ class BasicAgent:
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        model = InferenceClientModel(model_id="Qwen/Qwen2.5-72B-Instruct" )
         agent = CodeAgent(
         model=model,
         tools=[DuckDuckGoSearchTool(), VisitWebpageTool()],

         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
+        api_key = os.getenv("HF_TOKEN")
+        model = InferenceClientModel(api_key=api_key, model_id="Qwen/Qwen2.5-72B-Instruct" )
         agent = CodeAgent(
         model=model,
         tools=[DuckDuckGoSearchTool(), VisitWebpageTool()],