Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from huggingface_hub import InferenceClient, login
+from huggingface_hub import HfApi, InferenceClient, login
 
 
 # (Keep Constants as is)
@@ -16,27 +16,23 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 from huggingface_hub import InferenceClient, login
 
 class BasicAgent:
+    
     def __init__(self, hf_token: str):
         # Required for gated models
         login(token=hf_token, add_to_git_credential=False)
         
+        # Initialize HfApi client
+        self.hf_api = HfApi(token=hf_token)
+        
         # Verify model access
         try:
-            model_info = HfApi().model_info(
-                repo_id="meta-llama/Meta-Llama-3-70B-Instruct",
-                token=hf_token
+            model_info = self.hf_api.model_info(
+                repo_id="meta-llama/Meta-Llama-3-70B-Instruct"
             )
             if model_info.gated:
-                print("Verified model access")
+                print("Verified Llama-3 access")
         except Exception as e:
             raise RuntimeError(f"Model access denied: {str(e)}")
-
-        # Initialize client
-        self.client = InferenceClient(
-            model="meta-llama/Meta-Llama-3-70B-Instruct",
-            token=hf_token,
-            timeout=60
-        )
         
     def __call__(self, question: str) -> str:
         try: