Register all timm models with ImageNet-1K and update documentation (#24)

* register all timm models with imagenet 1k

* remove eva

* update readme

README.md

@@ -194,20 +194,7 @@ model = xinfer.create_model(model)
 
 
 TIMM:
-<table>
-    <thead>
-        <tr>
-            <th>Model</th>
-            <th>Usage</th>
-        </tr>
-    </thead>
-    <tbody>
-        <tr>
-            <td><a href="https://github.com/baaivision/EVA/tree/master/EVA-02">EVA02 Series</a></td>
-            <td><code>xinfer.create_model("eva02_small_patch14_336.mim_in22k_ft_in1k")</code></td>
-        </tr>
-    </tbody>
-</table>
+All models from [TIMM](https://github.com/huggingface/pytorch-image-models) fine-tuned for ImageNet 1k are supported.
 
 > [!NOTE]
 > Wish to load an unsupported model?

requirements.txt

@@ -5,4 +5,5 @@ einops
 pillow
 gradio
 itables
-pandas
+pandas
+numpy<2.0.0

xinfer/timm/__init__.py

@@ -1,2 +1,2 @@
-from .eva02 import EVA02
+from .load_timm_models import load_timm_models
 from .timm_model import TimmModel

xinfer/timm/load_timm_models.py

@@ -0,0 +1,16 @@
+import timm
+
+from ..model_registry import ModelInputOutput, register_model
+from .timm_model import TimmModel
+
+
+def load_timm_models():
+    model_list = timm.list_models("*1k*", pretrained=True)
+
+    for model_id in model_list:
+        register_model(model_id, "timm", ModelInputOutput.IMAGE_TO_CATEGORIES)(
+            TimmModel
+        )
+
+
+load_timm_models()