Gaoussin commited on
Commit
ab77b7f
·
verified ·
1 Parent(s): 5e470f8

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -34,7 +34,7 @@ print(f"Using device: {device}")
34
  # =====================
35
  # Charger le modèle et le tokenizer NLLB
36
  try:
37
- model_name = "Gaoussin/bamalingua-5"
38
  tokenizer = NllbTokenizer.from_pretrained(model_name)
39
  # Move model to the selected device (CPU or GPU)
40
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name).to(device)
 
34
  # =====================
35
  # Charger le modèle et le tokenizer NLLB
36
  try:
37
+ model_name = "Gaoussin/bamalingua-6"
38
  tokenizer = NllbTokenizer.from_pretrained(model_name)
39
  # Move model to the selected device (CPU or GPU)
40
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name).to(device)