update

src/display/utils.py

@@ -62,25 +62,25 @@ class ModelDetails:
 
 
 class ModelType(Enum):
-    PT = ModelDetails(name="pretrained", symbol="🟢")
-    FT = ModelDetails(name="fine-tuned", symbol="🔶")
-    IFT = ModelDetails(name="instruction-tuned", symbol="⭕")
-    RL = ModelDetails(name="RL-tuned", symbol="🟦")
-    Unknown = ModelDetails(name="", symbol="?")
+    vid_llm = ModelDetails(name="Video-LLM")
+    img_llm = ModelDetails(name="Image-LLM")
+    llm = ModelDetails(name="LLM")
+    others = ModelDetails(name="others")
+    Unknown = ModelDetails(name="")
 
     def to_str(self, separator=" "):
         return f"{self.value.symbol}{separator}{self.value.name}"
 
     @staticmethod
     def from_str(type):
-        if "fine-tuned" in type or "🔶" in type:
-            return ModelType.FT
-        if "pretrained" in type or "🟢" in type:
-            return ModelType.PT
-        if "RL-tuned" in type or "🟦" in type:
-            return ModelType.RL
-        if "instruction-tuned" in type or "⭕" in type:
-            return ModelType.IFT
+        if "Video-LLM" in type:
+            return ModelType.vid_llm
+        if "Image-LLM" in type:
+            return ModelType.img_llm
+        if "LLM" in type:
+            return ModelType.llm
+        if "others" in type:
+            return ModelType.others
         return ModelType.Unknown
 
 class WeightType(Enum):