Update app.py

app.py

@@ -5,11 +5,12 @@ import os
 
 class GradioInference():
     def __init__(self):
-        self.sizes = list(whisper._MODELS.keys())
+        self.sizes = ["base", "large"]
         self.langs = ["none"] + sorted(list(whisper.tokenizer.LANGUAGES.values()))
         self.current_size = "base"
         self.loaded_model = whisper.load_model(self.current_size)
 
+
     def __call__(self, file, lang, size, subs):
         path = file.name