oza75 commited on
Commit
8b2961b
·
verified ·
1 Parent(s): 90c70f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -21,9 +21,10 @@ else:
21
  logger.info("Using CPU for inference.")
22
 
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
  translator = pipeline("translation", model=model_checkpoint, revision=revision, device=device, max_length=512)
28
  logger.info("Translation pipeline initialized successfully.")
29
 
 
21
  logger.info("Using CPU for inference.")
22
 
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