Update app.py

app.py

@@ -224,7 +224,8 @@ class Translators:
     
     def nllb(self):
         tokenizer = AutoTokenizer.from_pretrained(self.model_name, src_lang=self.sl)
-        model = AutoModelForSeq2SeqLM.from_pretrained(self.model_name, device_map="auto")
+        # model = AutoModelForSeq2SeqLM.from_pretrained(self.model_name, device_map="auto", torch_dtype=torch.bfloat16)
+        model = AutoModelForSeq2SeqLM.from_pretrained(self.model_name)
         translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang=self.sl, tgt_lang=self.tl)
         translated_text = translator(self.input_text, max_length=512)
         return translated_text[0]['translation_text']