add

app.py

@@ -6,11 +6,12 @@ import logging
 # Set up logging
 logging.basicConfig(level=logging.INFO)
 
-def get_model_info(model_id="Qwen/Qwen2-7B-Instruct"):
+def get_model_info(model_id="Qwen/Qwen2-7B-Instruct", hf_token=None):
     url = f"https://huggingface.co/api/integrations/aws/v1/lookup/{model_id}"
+    headers = {"Authorization": f"Bearer {hf_token}"} if hf_token else {}
     logging.info(f"Requesting model info for model ID: {model_id}")
     
-    response = requests.get(url)
+    response = requests.get(url, headers=headers)
     
     if response.status_code != 200:
         logging.error(f"Error: {response.status_code} - {response.text}")
@@ -21,8 +22,8 @@ def get_model_info(model_id="Qwen/Qwen2-7B-Instruct"):
     return json.dumps(data, indent=4)
 
 iface = gr.Interface(
-    fn=get_model_info,
-    inputs=gr.Textbox(label="Model ID", placeholder="Enter model ID, e.g., HuggingFaceH4/zephyr-7b-beta"),
+    fn=lambda model_id: get_model_info(model_id, hf_token="YOUR_HF_TOKEN"),
+    inputs=gr.Textbox(label="Model ID", placeholder="HuggingFaceH4/zephyr-7b-beta"),
     outputs=gr.Textbox(label="API Response", lines=20),
     title="Hugging Face Model Lookup",
     description="Enter a model ID to retrieve its AWS integration details from Hugging Face."