Spaces:

themasterbetters
/

the-master-betters-translator

Running

dipesh1701 commited on Aug 10, 2023

Commit

5cac2b1

•

1 Parent(s): ccf30c7

bug fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -28,7 +28,7 @@ model_dict = load_models()
 # Translate text using preloaded models and tokenizers
 def translate_text(source, target, text):
-    model_name = "nllb-distilled-600M"
     if model_name in model_dict and model_dict[model_name]["model"] is not None:
         model = model_dict[model_name]["model"]

 # Translate text using preloaded models and tokenizers
 def translate_text(source, target, text):
+    model_name = "nllb-distilled-1.3B"
     if model_name in model_dict and model_dict[model_name]["model"] is not None:
         model = model_dict[model_name]["model"]