Update app.py

app.py

@@ -36,20 +36,29 @@ def extract_task(user_input):
 # 🔍 Scrape models from Hugging Face
 def get_models_for_task(task):
     url = f"https://huggingface.co/models?pipeline_tag={task}"
-    response = requests.get(url)
+    headers = {"User-Agent": "Mozilla/5.0"}  # avoid bot detection
+    response = requests.get(url, headers=headers)
+    
+    if response.status_code != 200:
+        raise Exception(f"Failed to fetch models: HTTP {response.status_code}")
+    
     soup = BeautifulSoup(response.text, "html.parser")
-    model_blocks = soup.select("div[data-testid='model-card']")
+    model_cards = soup.find_all("article")
 
     models_info = []
-    for block in model_blocks[:10]:  # limit to top 10 models
-        name = block.select_one("a[data-testid='model-link']")
-        arch = block.select_one("div[class*='tag']")  # very rough heuristic
+    for card in model_cards[:10]:  # Limit to top 10
+        name_tag = card.find("a", href=True)
+        tags = card.find_all("span", class_="tag") or card.find_all("div", class_="tag")
+
+        name = name_tag.text.strip() if name_tag else "unknown"
+        arch = tags[0].text.strip() if tags else "unknown"
 
         models_info.append({
-            "Model Name": name.text.strip() if name else "unknown",
+            "Model Name": name,
             "Task": task,
-            "Architecture": arch.text.strip() if arch else "unknown"
+            "Architecture": arch,
         })
+
     return models_info
 
 # 🎛 Gradio UI