Update app.py

app.py

@@ -18,7 +18,7 @@ print("Loading Translation Model... This may take a few minutes.")
 tokenizer_trans = AutoTokenizer.from_pretrained(MODEL_TRANS_ID)
 model_trans = AutoModelForCausalLM.from_pretrained(
     MODEL_TRANS_ID,
-    torch_dtype=torch.bfloat16,
+    dtype=torch.bfloat16,
     device_map="auto",
     trust_remote_code=True
 )
@@ -28,7 +28,7 @@ print("Loading Chat Model...")
 tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(
     MODEL_ID,
-    torch_dtype=torch.bfloat16,
+    dtype=torch.bfloat16,
     device_map="auto",
     trust_remote_code=True
 )