ts

app.py

@@ -8,30 +8,47 @@ import os
 logging.basicConfig(level=logging.INFO)
 
 def get_model_info(model_id="Qwen/Qwen2-7B-Instruct", hf_token=os.getenv('HF_TOKEN')):
-    url = f"https://huggingface.co/api/integrations/aws/v1/lookup/{model_id}"
-    headers = {
-        "Authorization": f"Bearer {hf_token}",
-        "Content-Type": "application/json",
-        "User-Agent": "Mozilla/5.0",  # Add a standard user agent
-        "Accept": "application/json"
+    url = f"https://huggingface.co/api/models"
+    params = {
+        "limit": 5,
+        "pipeline_tag": "text-generation",
+        "sort": "likes30d",
+        "full": "True",
+        "config": "True"
     }
+    
+    headers = {}
+    if hf_token:
+        headers["Authorization"] = f"Bearer {hf_token}"
+    try:
+        response = requests.get(url, params=params, headers=headers)
+    except requests.exceptions.RequestException as e:
+        print(f"Error fetching data for task text-generation: {e}")
+        return None
+    data = response.json()
+    return json.dumps(data, indent=4)
+    
+    # url = f"https://huggingface.co/api/integrations/aws/v1/lookup/{model_id}"
+    # headers = {
+    #     "Authorization": f"Bearer {hf_token}",
+    # }
 
-    logging.info(f"Requesting model info for model ID: {model_id}")
-    logging.info(f"URL: {url}")
+    # logging.info(f"Requesting model info for model ID: {model_id}")
+    # logging.info(f"URL: {url}")
 
-    response = requests.get(url, headers=headers)
+    # response = requests.get(url, headers=headers)
     
-    logging.info(f"Response Status: {response.status_code}")
-    logging.info(f"Response Headers: {response.headers}")
-    logging.info(f"Response Text: {response.text}")
+    # logging.info(f"Response Status: {response.status_code}")
+    # logging.info(f"Response Headers: {response.headers}")
+    # logging.info(f"Response Text: {response.text}")
 
-    if response.status_code != 200:
-        logging.error(f"Error: {response.status_code} - {response.text}")
-        return f"Error: {response.status_code}\\nResponse: {response.text}"
+    # if response.status_code != 200:
+    #     logging.error(f"Error: {response.status_code} - {response.text}")
+    #     return f"Error: {response.status_code}\\nResponse: {response.text}"
     
-    data = response.json()
-    logging.info("Successfully retrieved model info.")
-    return json.dumps(data, indent=4)
+    # data = response.json()
+    # logging.info("Successfully retrieved model info.")
+    # return json.dumps(data, indent=4)
 
 iface = gr.Interface(
     fn=lambda model_id: get_model_info(model_id, hf_token=os.getenv('HF_TOKEN')),