From f97df1c6b968771676a47a0168ff1c362831cee7 Mon Sep 17 00:00:00 2001
From: jm12138 <2286040843@qq.com>
Date: Sun, 13 Jun 2021 19:46:13 +0800
Subject: [PATCH] update __init__

---
 ppim/__init__.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/ppim/__init__.py b/ppim/__init__.py
index 731d5f2..5f99b4f 100644
--- a/ppim/__init__.py
+++ b/ppim/__init__.py
@@ -1 +1,27 @@
+import ppim.models as models
+
 from ppim.models import *
+from inspect import isfunction, isclass
+
+version = "1.1.0"
+models_dict = {}
+models_list = []
+
+
+for k, v in models.__dict__.items():
+    if isfunction(v):
+        model_name = k.split("_")[0]
+        if model_name not in models_dict:
+            models_dict[model_name] = [k]
+        else:
+            models_dict[model_name].append(k)
+    elif isclass(v):
+        models_list.append(k)
+
+
+def available():
+    print("The pretrained models list as follow:")
+    print("\n".join([str({k: v}) for k, v in models_dict.items()]))
+
+    print("The models support as follow:")
+    print("\n".join(str([item]) for item in models_list))