some changes

src/server/main.py

@@ -372,9 +372,9 @@ SUPPORTED_LANGUAGES = {
 
 # Translation Manager
 class TranslateManager:
-    def __init__(self, src_lang, tgt_lang, device_type=device):
+    def __init__(self, src_lang, tgt_lang, device_type=device, use_distilled=True):
         self.device_type = torch.device(device_type)
-        self.tokenizer, self.model = self.initialize_model(src_lang, tgt_lang)
+        self.tokenizer, self.model = self.initialize_model(src_lang, tgt_lang, use_distilled)
 
     def initialize_model(self, src_lang, tgt_lang, use_distilled=True):
         if src_lang.startswith("eng") and not tgt_lang.startswith("eng"):