Update app.py

app.py

@@ -8,7 +8,7 @@ tgt_lang="zu"
 
 # tokenizer = AutoTokenizer.from_pretrained(model_path, src_lang=src_lang)
 tokenizer = AutoTokenizer.from_pretrained(model_path)
-model = AutoModelForSeq2SeqLM.from_pretrained(model_path)#, load_in_4bit=True, device_map="auto")
+model = AutoModelForSeq2SeqLM.from_pretrained(model_path, use_safetensors=True)#, load_in_4bit=True, device_map="auto")
 
 def translate(text, num_beams=4, num_return_sequences=4):
   inputs = tokenizer(text, return_tensors="pt")