Update llm_node.py

llm_node.py

@@ -2,11 +2,11 @@ import os
 from huggingface_hub import InferenceClient
 
 def llm_node(question, search_result):
-    # Initialize the Hugging Face Inference Client
-    client = InferenceClient(
-        repo_id="HuggingFaceH4/zephyr-7b-beta",  # ✅ You can swap in another, e.g., mistralai/Mistral-7B-Instruct-v0.2
-        token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
-    )
+    # Initialize the client (no repo_id here!)
+    client = InferenceClient(token=os.getenv("HUGGINGFACEHUB_API_TOKEN"))
+
+    # Define the model you want to use
+    model_id = "HuggingFaceH4/zephyr-7b-beta"  # You can swap this with e.g., mistralai/Mistral-7B-Instruct-v0.2
 
     # Craft the prompt carefully
     prompt = f"""You are solving a GAIA benchmark evaluation question.
@@ -24,9 +24,10 @@ Here’s retrieved information:
 
 Your answer:"""
 
-    # Call the model
+    # Call the model (pass model ID here)
     response = client.text_generation(
-        prompt,
+        model=model_id,
+        prompt=prompt,
         max_new_tokens=500,
         temperature=0.1,
         top_p=0.95,