Typo fix for model names

main.py

@@ -8,8 +8,8 @@ from models.model_v1.wrapper import HybridDebertaWrapper
 from models.model_v2.wrapper import ModelV2Wrapper
 
 AVAILABLE_MODELS: Dict[str, BaseModelWrapper] = {
-    "hybrid_deberta": HybridDebertaWrapper(),
-    "model_v2":ModelV2Wrapper()
+    "Model V1": HybridDebertaWrapper(),
+    "Model V2":ModelV2Wrapper()
 }
 
 @asynccontextmanager

models/model_v1/wrapper.py

@@ -99,5 +99,5 @@ class HybridDebertaWrapper(BaseModelWrapper):
             "confidence": prob,
             "is_dementia": prob >= self.config['threshold'],
             "attention_map": attention_map,
-            "model_used": "hybrid_deberta"
+            "model_used": "hybrid deberta"
         }

models/model_v2/wrapper.py

@@ -160,5 +160,5 @@ class ModelV2Wrapper(BaseModelWrapper):
                 "pauses_ratio": ling_feats[5]
             },
             "key_segments": top_sentences,
-            "model_used": "model_v2"
+            "model_used": "Model v2"
         }