Update app.py

app.py

@@ -39,17 +39,37 @@ def get_model_list():
             model_name = model_element.find('span', {'x-test-search-response-title': True}).text.strip()
             size_elements = model_element.find_all('span', {'x-test-size': True})
 
+#            # Filtrer les modèles par taille
+#            for size_element in size_elements:
+#                size = size_element.text.strip()
+#                if size.endswith('m'):
+#                    # Tous les modèles en millions sont acceptés
+#                    model_list.append(f"{model_name}:{size}")
+#                elif size.endswith('b'):
+#                    # Convertir les modèles en milliards en milliards
+#                    size_value = float(size[:-1])
+#                    if size_value <= 10:  # Filtrer les modèles <= 10 milliards de paramètres
+#                        model_list.append(f"{model_name}:{size}")
+
+
+
+
             # Filtrer les modèles par taille
             for size_element in size_elements:
-                size = size_element.text.strip()
-                if size.endswith('m'):
+                size = size_element.text.strip().lower()  # Convertir en minuscules
+                if 'x' in size:
+                    # Exclure les modèles avec des tailles de type nXm ou nXb
+                    continue
+                elif size.endswith('m'):
                     # Tous les modèles en millions sont acceptés
                     model_list.append(f"{model_name}:{size}")
                 elif size.endswith('b'):
                     # Convertir les modèles en milliards en milliards
                     size_value = float(size[:-1])
                     if size_value <= 10:  # Filtrer les modèles <= 10 milliards de paramètres
-                        model_list.append(f"{model_name}:{size}")
+            model_list.append(f"{model_name}:{size}")
+
+
 
         return model_list
     else: