Update src/display/utils.py

src/display/utils.py

@@ -75,11 +75,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 "Reasoning-enhanced" 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