Update app.py

app.py

@@ -17,8 +17,8 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Models ---
 # Vision model for image analysis / OCR
-vision_model = ApiModel(repo_id="FireRedTeam/FireRed-OCR", max_new_tokens=2048, temperature=0.3)
-math_model = ApiModel(repo_id="Qwen/Qwen2.5-Math-1.5B", max_new_tokens=2048, temperature=0.3)
+vision_model = ApiModel(model_id="FireRedTeam/FireRed-OCR", max_new_tokens=2048, temperature=0.3)
+math_model = ApiModel(model_id="Qwen/Qwen2.5-Math-1.5B", max_new_tokens=2048, temperature=0.3)
 
 #define the state
 class AgentState(TypedDict):
@@ -236,7 +236,7 @@ Context gathered:
 """
     messages = [HumanMessage(content=prompt)]
     # Use the general model for final answer synthesis
-    general_model = ApiModel(repo_id="Qwen3.5-35B-A3B", max_new_tokens=2048, temperature=0.3)
+    general_model = ApiModel(model_id="Qwen3.5-35B-A3B", max_new_tokens=2048, temperature=0.3)
     response = general_model.invoke(messages)
     raw_response = response.content if hasattr(response, 'content') else str(response)