oza75 commited on
Commit
10e4994
·
verified ·
1 Parent(s): 8b2961b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -23,8 +23,10 @@ else:
23
  # Load the translation pipeline with the specified model and detected device
24
  #model_checkpoint = "oza75/bm-nllb-1.3B"
25
  #revision = "f55e4e7193eb95483b084248e5e4a91e28eb7dc2"
26
- model_checkpoint = "oza75/nllb-600M-mt-french-bambara"
27
- revision=None
 
 
28
  translator = pipeline("translation", model=model_checkpoint, revision=revision, device=device, max_length=512)
29
  logger.info("Translation pipeline initialized successfully.")
30
 
 
23
  # Load the translation pipeline with the specified model and detected device
24
  #model_checkpoint = "oza75/bm-nllb-1.3B"
25
  #revision = "f55e4e7193eb95483b084248e5e4a91e28eb7dc2"
26
+ model_checkpoint = "oza75/bm-nllb-600-01"
27
+ revision = None
28
+ # model_checkpoint = "oza75/nllb-600M-mt-french-bambara"
29
+ # revision=None
30
  translator = pipeline("translation", model=model_checkpoint, revision=revision, device=device, max_length=512)
31
  logger.info("Translation pipeline initialized successfully.")
32