Switch emojis

src/display/utils.py

@@ -65,9 +65,9 @@ class ModelDetails:
 
 
 class ModelType(Enum):
-    PT = ModelDetails(name="pretrained", symbol="🟢")
+    PT = ModelDetails(name="pretrained", symbol="⭕")
     FT = ModelDetails(name="fine-tuned", symbol="🔶")
-    IFT = ModelDetails(name="instruction-tuned", symbol="⭕")
+    IFT = ModelDetails(name="instruction-tuned", symbol="🟢")
     RL = ModelDetails(name="RL-tuned", symbol="🟦")
     Unknown = ModelDetails(name="", symbol="?")
 
@@ -78,11 +78,11 @@ class ModelType(Enum):
     def from_str(type):
         if "fine-tuned" in type or "🔶" in type:
             return ModelType.FT
-        if "pretrained" in type or "🟢" in type:
+        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:
+        if "instruction-tuned" in type or "🟢" in type:
             return ModelType.IFT
         return ModelType.Unknown