From 4dc633831dbf080cfb1cbd1b5e2e757d400a74a6 Mon Sep 17 00:00:00 2001
From: Daniel Ramos <46768340+danielcamposramos@users.noreply.github.com>
Date: Sun, 26 Jan 2025 21:49:07 -0300
Subject: [PATCH] Update models.py

Changed to more modern and bigger models from fastwhisper;
---
 transcription-api/models.py | 16 +++++-----------
 1 file changed, 5 insertions(+), 11 deletions(-)

diff --git a/transcription-api/models.py b/transcription-api/models.py
index cbfeceb..1b88fe8 100644
--- a/transcription-api/models.py
+++ b/transcription-api/models.py
@@ -6,16 +6,10 @@ class DeviceType(str, Enum):
     cuda = "cuda"
 
 class ModelSize(str, Enum):
-    tiny_en = "tiny.en"
-    tiny = "tiny"
-    base_en = "base.en"
-    base = "base"
-    small_en = "small.en"
-    small = "small"
-    medium_en = "medium.en"
-    medium = "medium"
-    large_v2 = "large-v2"
-    large_v3 = "large-v3"
+    distil_large_v3 = "distil-large-v3"
+    distil_large_v2 = "distil-large-v2"
+    distil_medium = "distil-medium.en"
+    turbo = "turbo"
 
 class Languages(str, Enum):
     auto = "auto"
@@ -43,4 +37,4 @@ class Languages(str, Enum):
     sv = "sv"
     tk = "tk"
     tr = "tr"
-    zh = "zh"
\ No newline at end of file
+    zh = "zh"