Added GPU Models

models.py

@@ -19,15 +19,15 @@ def load_models():
         models = dict()
 
         # Transcription Model - Whisper
-        # models['transcription'] = stable_whisper.load_model('large-v2')    # Stable Whisper
+        models['transcription'] = stable_whisper.load_model('large-v2')    # Stable Whisper
         # models['transcription'] = WhisperModel("large-v2", device="cuda", compute_type="float16")    # Faster Whisper
 
         # Translation Model - NLLB
-        # nllb_model = 'facebook/nllb-200-distilled-600M'
+        nllb_model = 'facebook/nllb-200-distilled-600M'
         # nllb_model = 'facebook/nllb-200-1.3B'
         # nllb_model = 'facebook/nllb-200-3.3B'
         # nllb_model = 'facebook/nllb-moe-54b'
-        # models['translation'] = dlt.TranslationModel(nllb_model)
+        models['translation'] = dlt.TranslationModel(nllb_model)
         
         # TODO: Audio Generation Model - Bark
         # models['audiobook'] =